diff --git a/cognitive_game_vars.py b/cognitive_game_vars.py
index 3571ef6dcd2accbedbd8fb00851c12d4e4b54a40..4fa0f4bdf95e9a846d9733005ff5b20d40a49a2b 100644
--- a/cognitive_game_vars.py
+++ b/cognitive_game_vars.py
@@ -43,7 +43,7 @@ class Attention(enum.Enum):
     name = "attention"
     counter = 3
 
-class Robot_Assistance(enum.Enum):
+class Agent_Assistance(enum.Enum):
     lev_0 = 0
     lev_1 = 1
     lev_2 = 2
@@ -53,7 +53,7 @@ class Robot_Assistance(enum.Enum):
     name = "robot_assistance"
     counter = 6
 
-class Robot_Feedback(enum.Enum):
+class Agent_Feedback(enum.Enum):
     yes = 1
     no = 0
     name = "robot_feedback"
diff --git a/episode.py b/episode.py
index cdf63e82f7369e4ed0ea6fb042d0a9ab602eae9b..bcc420a4700d063be951b53981ced05390cb313a 100644
--- a/episode.py
+++ b/episode.py
@@ -74,8 +74,8 @@ class Episode:
         '''
         It returns the episodes related to the saved file
         :param file:
-        :param episode: look at main.py
-        :param sol_per_pop: look at main.py
+        :param episode: look at simulation.py
+        :param sol_per_pop: look at simulation.py
         :return: a list of episodes
         '''
         print("LOADING...")
@@ -167,7 +167,7 @@ class Episode:
 
 def main():
     pass
-    # file_path = "/home/aandriella/Documents/Codes/MY_FRAMEWORK/BN_GenerativeModel/results/1/episodes.npy"
+    # file_path = "/home/aandriella/Documents/Codes/MY_FRAMEWORK/bn_generative_model/results/1/episodes.npy"
     # ep = Episode()
     # episodes = ep.load_episodes(file_path)
     # initial_state = (1, 1, 0)
@@ -187,8 +187,8 @@ def main():
     #                              task_length, n_max_attempt, timeout=0, n_levels_assistance=6)
     # #
     # trans_matrix = ep.generate_statistics(env.states, env.action_space, episodes)
-    # path_trans_matrix_occ = "/home/aandriella/Documents/Codes/MY_FRAMEWORK/BN_GenerativeModel/results/1/trans_matrix_occ.npy"
-    # path_trans_matrix_prob = "/home/aandriella/Documents/Codes/MY_FRAMEWORK/BN_GenerativeModel/results/1/trans_matrix_prob.npy"
+    # path_trans_matrix_occ = "/home/aandriella/Documents/Codes/MY_FRAMEWORK/bn_generative_model/results/1/trans_matrix_occ.npy"
+    # path_trans_matrix_prob = "/home/aandriella/Documents/Codes/MY_FRAMEWORK/bn_generative_model/results/1/trans_matrix_prob.npy"
     # terminal_states = [env.point_to_index(state) for state in final_states]
     #
     #