diff --git a/RLLib_training/render_training_result.py b/RLLib_training/render_training_result.py
index 90b38a3329b47163a4f22f61b39d81b0f1ab26ac..668c763d162bbafd1bb5c62fb2317324998768f3 100644
--- a/RLLib_training/render_training_result.py
+++ b/RLLib_training/render_training_result.py
@@ -32,7 +32,7 @@ ray.init()  # object_store_memory=150000000000, redis_max_memory=30000000000)
 __file_dirname__ = os.path.dirname(os.path.realpath(__file__))
 
 CHECKPOINT_PATH = os.path.join(__file_dirname__, 'experiment_configs', 'config_example', 'ppo_policy_two_obs_with_predictions_n_agents_4_map_size_20q58l5_f7',
-                               'checkpoint_101', 'checkpoint-101')
+                               'checkpoint_101', 'checkpoint-101')  # To Modify
 N_EPISODES = 10
 N_STEPS_PER_EPISODE = 50
 
@@ -165,5 +165,5 @@ def run_experiment(name, num_iterations, n_agents, hidden_sizes, save_every,
 
 
 if __name__ == '__main__':
-    gin.parse_config_file(os.path.join(__file_dirname__, 'experiment_configs', 'config_example', 'config.gin'))
+    gin.parse_config_file(os.path.join(__file_dirname__, 'experiment_configs', 'config_example', 'config.gin'))  # To Modify
     run_experiment()
diff --git a/RLLib_training/train_experiment.py b/RLLib_training/train_experiment.py
index e7e9709aa092ebd2b886aa005731e386f5f2e89c..808bbc34556ce541ca41fc560b464c31b1dd96a3 100644
--- a/RLLib_training/train_experiment.py
+++ b/RLLib_training/train_experiment.py
@@ -179,7 +179,7 @@ def run_experiment(name, num_iterations, n_agents, hidden_sizes, save_every,
                 "kl_coeff": kl_coeff,
                 "lambda_gae": lambda_gae,
                 "min_dist": min_dist,
-                "step_memory": step_memory
+                    "step_memory": step_memory
                 },
         resources_per_trial={
             "cpu": 3,
@@ -191,7 +191,7 @@ def run_experiment(name, num_iterations, n_agents, hidden_sizes, save_every,
 
 
 if __name__ == '__main__':
-    print(str(os.path.join(__file_dirname__, 'experiment_configs', 'config_example', 'config.gin')))
-    gin.parse_config_file(os.path.join(__file_dirname__, 'experiment_configs', 'config_example', 'config.gin'))
-    dir = os.path.join(__file_dirname__, 'experiment_configs', 'config_example')
+    folder_name = 'config_example'  # To Modify
+    gin.parse_config_file(os.path.join(__file_dirname__, 'experiment_configs', folder_name, 'config.gin'))
+    dir = os.path.join(__file_dirname__, 'experiment_configs', folder_name)
     run_experiment(local_dir=dir)