diff --git a/src/AGENT/GlobalVar.py b/src/AGENT/GlobalVar.py
index 85fefa4..a6a44ac 100644
--- a/src/AGENT/GlobalVar.py
+++ b/src/AGENT/GlobalVar.py
@@ -30,7 +30,7 @@
 # LABELS_TO_TRAIN = new_LABELS
 
 
-
+LABELS_TO_TRAIN = ['RAF','RAE']
 
 LABEL_GROUPES = {}
 LABELS = []
diff --git a/src/AGENT/predic_landmarks.py b/src/AGENT/predict_landmarks.py
similarity index 96%
rename from src/AGENT/predic_landmarks.py
rename to src/AGENT/predict_landmarks.py
index c80207f..3a05927 100644
--- a/src/AGENT/predic_landmarks.py
+++ b/src/AGENT/predict_landmarks.py
@@ -133,7 +133,7 @@ def main(args):
 
 
     # agent_lst = GetAgentLst(agents_param)
-    brain_lst = GetBrain(args.dir_model)
+    brain_lst = GetBrain(args.dir_models)
     # print( brain_lst)
     # environement_lst = [environement_lst[0]]
     # agent_lst = [agent_lst[0]]
@@ -200,9 +200,9 @@ def main(args):
     
     input_group = parser.add_argument_group('dir')
     input_group.add_argument('--dir_scans', type=str, help='Input directory with the scans',default='/Users/luciacev-admin/Documents/Projects/MSDRL_benchmark/data/test')
-    input_group.add_argument('--dir_model', type=str, help='Directory of the trained models',default= '/Users/luciacev-admin/Desktop/MSDRL_models/ALI_CNN_models_2021_26_10')
+    input_group.add_argument('--dir_models', type=str, help='Directory of the trained models',default= '/Users/luciacev-admin/Desktop/MSDRL_models/ALI_CNN_models_2021_26_10')
     
-    input_group.add_argument('--clear_temp', type=bool, help='Temp directory',default= False)
+    input_group.add_argument('--clear_temp', type=bool, help='Temp directory',default= True)
     input_group.add_argument('--dir_temp', type=str, help='Temp directory',default= '..')