diff --git a/torch_training/multi_agent_training.py b/torch_training/multi_agent_training.py
index 0862e30e1761375bcb35026ea2ec4cfe74886df5..bee0d96ec6c530ac3007e03c77301c7fc92d2862 100644
--- a/torch_training/multi_agent_training.py
+++ b/torch_training/multi_agent_training.py
@@ -44,6 +44,8 @@ def main(argv):
     print("main2")
 
     # Get an observation builder and predictor
+    # The predictor will always predict the shortest path from the current location of the agent.
+    # This is used to warn for potential conflicts --> Should be enhanced to get better performance!
     predictor = ShortestPathPredictorForRailEnv()
     observation_helper = TreeObsForRailEnv(max_depth=tree_depth, predictor=predictor)