diff --git a/evaluation.py b/evaluation.py
index c4ea784..aea3416 100644
--- a/evaluation.py
+++ b/evaluation.py
@@ -40,8 +40,8 @@ with open('eval.csv', 'r') as r:
 
 
 plt.ylabel('F score')
-plt.plot(np.arange(0, len(MAE)), MAE)
-plt.plot(np.arange(0, len(MSE)), MSE) 
-plt.plot(np.arange(0, len(RMSE)), RMSE) 
+plt.plot(np.arange(0, len(MAE)), MAE, label="MAE")
+plt.plot(np.arange(0, len(MSE)), MSE, label="MSE") 
+plt.plot(np.arange(0, len(RMSE)), RMSE, label="RMSE") 
 
 plt.savefig('metrics.png')
\ No newline at end of file