diff --git a/part4/eval.py b/part4/eval.py
index e9369e7..7c13310 100644
--- a/part4/eval.py
+++ b/part4/eval.py
@@ -10,7 +10,7 @@ def eval():
 
     # Re-train 10 times in order to reduce effects of randomness
     for i in range(10):
-        ### TODO: Exercise 5
+        ### TODO: Exercise 4-1
         ### 1. Load data from traces file
         ### 2. Split data into X_train, X_test, y_train, y_test with train_test_split
         ### 3. Train classifier with X_train and y_train
@@ -20,7 +20,7 @@ def eval():
         y_test_full.extend(y_test)
         y_pred_full.extend(y_pred)
 
-    ### TODO: Exercise 5 (continued)
+    ### TODO: Exercise 4-1 (continued)
     ### 5. Print classification report using y_test_full and y_pred_full
 
 if __name__ == "__main__":