diff --git a/torch_training/multi_agent_training.py b/torch_training/multi_agent_training.py
index bee0d96ec6c530ac3007e03c77301c7fc92d2862..4d337d1d8efa3bb455eeb9ecbd06ca783bafa673 100644
--- a/torch_training/multi_agent_training.py
+++ b/torch_training/multi_agent_training.py
@@ -43,9 +43,11 @@ def main(argv):
     tree_depth = 3
     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!
+    """
+     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)