diff --git a/batch_train.sh b/batch_train.sh
index 4b4036e..9d27dec 100644
--- a/batch_train.sh
+++ b/batch_train.sh
@@ -3,45 +3,45 @@
 echo 
 
 echo ddpg_ssr
-python3 main_test.py --drl ddpg --reward ssr --ep-num 300 --trained-uav
+python3 main_train.py --drl ddpg --reward ssr --ep-num 300 --trained-uav
 echo ddpg_ssr_2
-python3 main_test.py --drl ddpg --reward ssr --ep-num 300 --trained-uav
+python3 main_train.py --drl ddpg --reward ssr --ep-num 300 --trained-uav
 echo ddpg_ssr_3
-python3 main_test.py --drl ddpg --reward ssr --ep-num 300 --trained-uav
+python3 main_train.py --drl ddpg --reward ssr --ep-num 300 --trained-uav
 echo ddpg_ssr_4
-python3 main_test.py --drl ddpg --reward ssr --ep-num 300 --trained-uav
+python3 main_train.py --drl ddpg --reward ssr --ep-num 300 --trained-uav
 echo ddpg_ssr_5
-python3 main_test.py --drl ddpg --reward ssr --ep-num 300 --trained-uav
+python3 main_train.py --drl ddpg --reward ssr --ep-num 300 --trained-uav
 
 echo td3_ssr
-python3 main_test.py --drl td3 --reward ssr --ep-num 300 --trained-uav
+python3 main_train.py --drl td3 --reward ssr --ep-num 300 --trained-uav
 echo td3_ssr_2
-python3 main_test.py --drl td3 --reward ssr --ep-num 300 --trained-uav
+python3 main_train.py --drl td3 --reward ssr --ep-num 300 --trained-uav
 echo td3_ssr_3
-python3 main_test.py --drl td3 --reward ssr --ep-num 300 --trained-uav
+python3 main_train.py --drl td3 --reward ssr --ep-num 300 --trained-uav
 echo td3_ssr_4
-python3 main_test.py --drl td3 --reward ssr --ep-num 300 --trained-uav
+python3 main_train.py --drl td3 --reward ssr --ep-num 300 --trained-uav
 echo td3_ssr_5
-python3 main_test.py --drl td3 --reward ssr --ep-num 300 --trained-uav
+python3 main_train.py --drl td3 --reward ssr --ep-num 300 --trained-uav
 
 echo ddpg_see
-python3 main_test.py --drl ddpg --reward see --ep-num 300 --trained-uav
+python3 main_train.py --drl ddpg --reward see --ep-num 300 --trained-uav
 echo ddpg_see_2
-python3 main_test.py --drl ddpg --reward see --ep-num 300 --trained-uav
+python3 main_train.py --drl ddpg --reward see --ep-num 300 --trained-uav
 echo ddpg_see_3
-python3 main_test.py --drl ddpg --reward see --ep-num 300 --trained-uav
+python3 main_train.py --drl ddpg --reward see --ep-num 300 --trained-uav
 echo ddpg_see_4
-python3 main_test.py --drl ddpg --reward see --ep-num 300 --trained-uav
+python3 main_train.py --drl ddpg --reward see --ep-num 300 --trained-uav
 echo ddpg_see_5
-python3 main_test.py --drl ddpg --reward see --ep-num 300 --trained-uav
+python3 main_train.py --drl ddpg --reward see --ep-num 300 --trained-uav
 
 echo td3_see
-python3 main_test.py --drl td3 --reward see --ep-num 300 --trained-uav
+python3 main_train.py --drl td3 --reward see --ep-num 300 --trained-uav
 echo td3_see_2
-python3 main_test.py --drl td3 --reward see --ep-num 300 --trained-uav
+python3 main_train.py --drl td3 --reward see --ep-num 300 --trained-uav
 echo td3_see_3
-python3 main_test.py --drl td3 --reward see --ep-num 300 --trained-uav
+python3 main_train.py --drl td3 --reward see --ep-num 300 --trained-uav
 echo td3_see_4
-python3 main_test.py --drl td3 --reward see --ep-num 300 --trained-uav
+python3 main_train.py --drl td3 --reward see --ep-num 300 --trained-uav
 echo td3_see_5
-python3 main_test.py --drl td3 --reward see --ep-num 300 --trained-uav
+python3 main_train.py --drl td3 --reward see --ep-num 300 --trained-uav