diff --git a/src/primaite/agents/rllib.py b/src/primaite/agents/rllib.py
index 32dc3dc0..ce072a03 100644
--- a/src/primaite/agents/rllib.py
+++ b/src/primaite/agents/rllib.py
@@ -186,6 +186,7 @@ class RLlibAgent(AgentSessionABC):
             checkpoint_dir = file
             break
 
+        # Zip the folder
         shutil.make_archive(
             str(self._saved_agent_path).replace(".zip", ""),
             "zip",