diff --git a/DG_Approximation.py b/DG_Approximation.py
index aa9408a8900ef755ab606b05d87ca3ac17836f7f..fc43cdb04b34e9d804b39afb524af6185feeea4a 100644
--- a/DG_Approximation.py
+++ b/DG_Approximation.py
@@ -5,10 +5,11 @@
 Plotter:
 TODO: Double-check everything!
 TODO: Replace loops with list comprehension if feasible
-TODO: Write documentation for all methods
+TODO: Write documentation for all methods (important)
 TODO: Contemplate how to make shock tubes comparable
 TODO: Check whether 'projection' is always a np.array()
 TODO: Check whether all instance variables sensible
+TODO: Use cfl_number for updating, not just time
 
 """
 import numpy as np
diff --git a/Troubled_Cell_Detector.py b/Troubled_Cell_Detector.py
index 5e89b9e7fc5105d55db9044db86e813228ec6d15..15c32b0ff28a46ce407f66534a18e8c7dd4e86c9 100644
--- a/Troubled_Cell_Detector.py
+++ b/Troubled_Cell_Detector.py
@@ -2,9 +2,7 @@
 """
 @author: Laura C. Kühle, Soraya Terrab (sorayaterrab)
 
-TODO: Fix cell averages and reconstructions to create data with an x-point stencil -> Done
-TODO: Add comments to get_cells() for ArtificialNeuralNetwork -> Done
-TODO:
+TODO: Move plotting to separate file (try to adjust for different equations)
 
 """
 import os