diff --git a/DDPG_solver/main.py b/DDPG_solver/main.py
index 0bd634a..8f19591 100644
--- a/DDPG_solver/main.py
+++ b/DDPG_solver/main.py
@@ -1,3 +1,4 @@
+from env import PartitionMazeEnv
 from utils import str2bool, evaluate_policy
 from datetime import datetime
 from DDPG import DDPG_agent
@@ -6,9 +7,9 @@ import os
 import shutil
 import argparse
 import torch
+
 import sys
 sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
-from env import PartitionMazeEnv
 
 '''Hyperparameter Setting'''
 parser = argparse.ArgumentParser()
@@ -52,9 +53,10 @@ print(opt)
 
 
 def main():
-    EnvName = ['Pendulum-v1', 'LunarLanderContinuous-v2', 'Humanoid-v4',
+    EnvName = ['PartitionMaze_DDPG', 'Pendulum-v1', 'LunarLanderContinuous-v2', 'Humanoid-v4',
                'HalfCheetah-v4', 'BipedalWalker-v3', 'BipedalWalkerHardcore-v3']
-    BrifEnvName = ['PV1', 'LLdV2', 'Humanv4', 'HCv4', 'BWv3', 'BWHv3']
+    BrifEnvName = ['PM_DDPG', 'PV1', 'LLdV2',
+                   'Humanv4', 'HCv4', 'BWv3', 'BWHv3']
 
     # Build Env
     # env = gym.make(EnvName[opt.EnvIdex], render_mode = "human" if opt.render else None)