diff --git a/ANN_Training_Data_Generator.py b/ANN_Training_Data_Generator.py
index 805b5a44dc0888597b1fe28ae9b9bd72b2117236..8f6114eef952edb4d8d3186f8abc92393b7973eb 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: Fix type for log prints -> Done
 
 """
 
diff --git a/Artificial_Neural_Network.py b/Artificial_Neural_Network.py
index 24cee969086c2b537318b48865310d072a1cc0a1..b093b5560a10ffd03d81e4e732666eb0cb95de82 100644
--- a/Artificial_Neural_Network.py
+++ b/Artificial_Neural_Network.py
@@ -1,3 +1,12 @@
+# -*- coding: utf-8 -*-
+"""
+@author: Laura C. Kühle, Soraya Terrab (sorayaterrab)
+
+TODO: Adapt '_train()' to fit style
+TODO: Add ANN testing from Soraya
+TODO: Add ANN classification from Soraya
+
+"""
 import numpy as np
 import torch
 from torch.utils.data import TensorDataset, DataLoader
diff --git a/DG_Approximation.py b/DG_Approximation.py
index 9d09116bc8e004b7b7a7413b210c4c5ccbf8f5a9..56edebd278a47d3036fc72724e9309aa4d50cfca 100644
--- a/DG_Approximation.py
+++ b/DG_Approximation.py
@@ -7,9 +7,7 @@ TODO: Double-check everything!
 TODO: Replace loops with list comprehension if feasible
 TODO: Write documentation for all methods
 TODO: Contemplate how to make shock tubes comparable
-TODO: Fix bug in approximation -> Done (used num_grid_cells instead of cell_len for cfl_number in last step)
-TODO: Include check_wavelet() for all cases in build_training_data() -> Done
-TODO: Remove 'Custom' class in Quadrature -> Done
+TODO: Check whether 'projection' is always a np.array()
 
 """
 import numpy as np
diff --git a/Troubled_Cell_Detector.py b/Troubled_Cell_Detector.py
index dc060deec265f72dea744292228c8ab73510de37..72042bf5feb36e034e26ac4e5ded119ec47cc866 100644
--- a/Troubled_Cell_Detector.py
+++ b/Troubled_Cell_Detector.py
@@ -2,13 +2,6 @@
 """
 @author: Laura C. Kühle, Soraya Terrab (sorayaterrab)
 
-TODO: Adapt 'train()' to fit style
-TODO: Add ANN testing from Soraya to ANN implementation
-TODO: Add ANN classification from Soraya to ANN implementation
-TODO: Move ANN implementation to new file -> Done (Artificial_Neural_Network)
-TODO: Implement 'get_cells()' for 'ArtificialNeuralNetwork' -> Done
-TODO: Adapt calculate_approximate_solution() to not require a quadrature -> Done
-TODO: Add function to determine cell average and reconstructions -> Done
 TODO: Fix cell averages and reconstructions to create data with an x-point stencil
 
 """