diff --git a/ANN_Training.py b/ANN_Training.py
index 209db3d9421f44f0464d431fe9bf240bb734534f..0f3afb7c3c0cb81e874a3c17d3dc923f4690e2dc 100644
--- a/ANN_Training.py
+++ b/ANN_Training.py
@@ -3,14 +3,11 @@
 @author: Laura C. Kühle, Soraya Terrab (sorayaterrab)
 
 TODO: Give option to compare multiple models
-TODO: Use sklearn for classification -> Done
-TODO: Fix difference between accuracies (stems from rounding; choose higher value instead) -> Done
 TODO: Add more evaluation measures (AUROC, ROC, F1, training accuracy, boxplot over CVF, etc.)
 TODO: Add log to pipeline
 TODO: Remove object set-up
 TODO: Optimize Snakefile-vs-config relation
 TODO: Improve maximum selection runtime
-TODO: Fix tensor mapping warning -> Done
 
 """
 import numpy as np