diff --git a/evaluation.py b/evaluation.py
index bb2a898..0e34c72 100644
--- a/evaluation.py
+++ b/evaluation.py
@@ -21,5 +21,4 @@ scores = model.evaluate(x=test_X,
                     y=test_Y)    
 
 with open('rmse.txt', 'w') as file:
-    for idx, score in enumerate(scores):
-        file.write(str(score) + "\n")
+    file.write(str(scores[1]) + "\n")
diff --git a/rmse.txt b/rmse.txt
new file mode 100644
index 0000000..01c2c36
--- /dev/null
+++ b/rmse.txt
@@ -0,0 +1,2 @@
+0.10223432630300522
+0.12947769463062286