diff --git a/ANN_Data_Generator.py b/ANN_Data_Generator.py
index 5b476f125c524d2b3aa0399a817b55daa5fb7d08..8aff0057111d7b5b445187b3876a3ec564343637 100644
--- a/ANN_Data_Generator.py
+++ b/ANN_Data_Generator.py
@@ -4,7 +4,6 @@
 
 TODO: Improve '_generate_cell_data'
 TODO: Extract normalization (Combine smooth and troubled before normalizing)
-TODO: Discontinue data splitting -> Done
 TODO: Improve verbose output
 
 """
diff --git a/ANN_Training.py b/ANN_Training.py
index 17de35092ab485c887f433b23b33cffe7393f8fd..dccd8b1dcce91c5e3bf2f967b361a3d0d2b2363e 100644
--- a/ANN_Training.py
+++ b/ANN_Training.py
@@ -2,20 +2,13 @@
 """
 @author: Laura C. Kühle, Soraya Terrab (sorayaterrab)
 
-TODO: Improve 'epoch_training()' -> Done
 TODO: Add ANN classification from Soraya
-TODO: Improve naming of training data/model (maybe different folders?; split training data later;
-    total amount + percentage?) -> Done
 TODO: Give option to compare multiple models
 TODO: Use sklearn for classification
 TODO: Fix difference between accuracies (stems from rounding; choose higher value instead)
 TODO: Add more evaluation measures (AUROC, ROC, F1, training accuracy, etc.)
 TODO: Decide on k-fold cross-validation (Use? Which model do we keep?)
-TODO: Fix bug in valid_loss calculation -> Done
-TODO: Replace multi-file set-up for training data -> Done
 TODO: Rework model testing
-TODO: Limit batch size with regard to training data -> Done
-TODO: Build Snakemake pipeline for model training -> Done
 TODO: Clean up directories/naming
 TODO: Add log to pipeline