diff --git a/program.py b/program.py
index 2dd42fc..56f743e 100644
--- a/program.py
+++ b/program.py
@@ -16,7 +16,7 @@ with open("train/in.tsv") as f:
 model = Word2Vec(X_train, min_count=1,size= 50,workers=3, window =3, sg = 1)
 X_train=model[X_train]
 X=model[X]
-FEAUTERES=10000
+FEAUTERES=1000
 class NeuralNetworkModel(torch.nn.Module):
 
     def __init__(self):
@@ -60,7 +60,7 @@ for epoch in range(10):
 
 
         loss_score += loss.item() * Y.shape[0] 
-        with open('test-A/out1.tsv', 'w') as file:
+        with open('test-A/out.tsv', 'w') as file:
             for e in Y_predictions:
                 file.write("%f\n" % e)