diff --git a/ANN_Training_Data_Generator.py b/ANN_Training_Data_Generator.py
index f81cce1804f31dc0b9ef94efe965e04daefd4e8a..2daf59fc1ae82d7da4e7f8053a25a71a1d36d31f 100644
--- a/ANN_Training_Data_Generator.py
+++ b/ANN_Training_Data_Generator.py
@@ -3,7 +3,6 @@
 @author: Soraya Terrab (sorayaterrab), Laura C. Kühle
 
 TODO: Improve '_generate_cell_data'
-TODO: Change set name from 'validation' to 'valid' -> Done
 
 """
 
diff --git a/Artificial_Neural_Network.py b/Artificial_Neural_Network.py
index 29767d2c8b29304d0d76bf2fd6a6ef18dbe94523..932df6e1dba2568dc7ee6b8b5e23ea825bd1efd0 100644
--- a/Artificial_Neural_Network.py
+++ b/Artificial_Neural_Network.py
@@ -3,22 +3,9 @@
 @author: Laura C. Kühle, Soraya Terrab (sorayaterrab)
 
 TODO: Improve 'epoch_training()'
+TODO: Improve 'save_model()'
 TODO: Add ANN testing from Soraya
 TODO: Add ANN classification from Soraya
-TODO: Implement __init__() -> Done
-TODO: Add reading of training data -> Done
-TODO: Add option to set batch size -> Done
-TODO: Rename '_train()' -> Done ('epoch_training()')
-TODO: Remove class declarations from 'epoch_training()' -> Done
-TODO: Move setting of training data in separate function -> Done ('_read_training_data()')
-TODO: Rename 'ArtificialNeuralNetwork' class -> Done ('ModelTrainer')
-TODO: Move ANN models in separate file -> Done ('ANN_Model.py')
-TODO: Take model as class input -> Done
-TODO: Take loss function as class input -> Done
-TODO: Take optimizer as class input -> Done
-TODO: Take parameter as class input -> Done
-TODO: Move saving in separate function -> Done ('save_model()')
-TODO: Improve 'save_model()'
 
 """
 import numpy as np
diff --git a/Troubled_Cell_Detector.py b/Troubled_Cell_Detector.py
index 096dc982771510a9761d6d5e85e1b355009d31b9..b83b735591087d6ffb7f5d42c12c41c8cbd85266 100644
--- a/Troubled_Cell_Detector.py
+++ b/Troubled_Cell_Detector.py
@@ -3,7 +3,6 @@
 @author: Laura C. Kühle, Soraya Terrab (sorayaterrab)
 
 TODO: Fix cell averages and reconstructions to create data with an x-point stencil
-TODO: Only allow model from 'ANN_Model' -> Done
 
 """
 import os