diff --git a/src/stratigraphy/util/predictions.py b/src/stratigraphy/util/predictions.py
index 0052b61b..2f1a311b 100644
--- a/src/stratigraphy/util/predictions.py
+++ b/src/stratigraphy/util/predictions.py
@@ -10,7 +10,10 @@
 
 
 class MaterialDescriptionPrediction:
-    """A class to represent a material description prediction."""
+    """A class to represent a material description prediction.
+
+    TODO: Check if this class can be replaced.
+    """
 
     def __init__(self, text: str, rect: list, lines: list):
         self.text = text
@@ -19,7 +22,10 @@ def __init__(self, text: str, rect: list, lines: list):
 
 
 class DepthIntervalPrediction:
-    """A class to represent a depth interval prediction."""
+    """A class to represent a depth interval prediction.
+
+    TODO: Could be replaced by BoundaryInterval from interval.py. Then we could use line_anchor and background_rect.
+    """
 
     def __init__(self, start: dict, end: dict):
         self.start = (