diff --git a/main.py b/main.py
index 93ba667..acfb17c 100644
--- a/main.py
+++ b/main.py
@@ -25,9 +25,5 @@ classifier = RandomForestClassifier()
 classifier.fit(X_train, y_transformed)
 y_pred = classifier.predict(X_test)
 
-# for i, prediction in enumerate(y_pred):
-#     print(data_test.iloc[i, 0] + ": ")
-#     print(prediction)
-
 data_test['8'] = list(y_pred)
-data_test.to_csv('prediction', sep=' ', index=False)
\ No newline at end of file
+data_test.to_csv('prediction.dat', sep=' ', index=False)
\ No newline at end of file