diff --git a/.gitignore b/.gitignore
index f3b16c0..cf3db71 100644
--- a/.gitignore
+++ b/.gitignore
@@ -224,3 +224,9 @@ poetry.toml
 pyrightconfig.json
 
 # End of https://www.toptal.com/developers/gitignore/api/python
+
+
+
+outputs/
+data/
+param_data/
diff --git a/README.md b/README.md
index ee01c1c..18789b2 100644
--- a/README.md
+++ b/README.md
@@ -35,7 +35,7 @@ git clone https://github.com/NUS-HPC-AI-Lab/Neural-Network-Diffusion.git
 2. Create a new Conda environment and activate it: 
 
 ```
-conda env create -f environment.yaml
+conda env create -f environment.yml
 conda activate pdiff
 ```
 
diff --git a/cifar100_resnet18_k200.sh b/cifar100_resnet18_k200.sh
old mode 100644
new mode 100755
index 0161d82..837545b
--- a/cifar100_resnet18_k200.sh
+++ b/cifar100_resnet18_k200.sh
@@ -1,2 +1,2 @@
-python task_training # get the model data for training p-diff
-python train_p_diff --task cifar100 --system ae_ddpm 
\ No newline at end of file
+python task_training.py # get the model data for training p-diff
+python train_p_diff.py task=cifar100 system=ae_ddpm 
\ No newline at end of file
diff --git a/configs/base.yaml b/configs/base.yaml
index a4be944..12853da 100644
--- a/configs/base.yaml
+++ b/configs/base.yaml
@@ -5,7 +5,7 @@ defaults:
 
 device:
   cuda_visible_devices:  "0,1,2,3,4,5,6"
-  id: 1
+  id: 0
   cuda: cuda:${device.id}
 
 load_system_checkpoint:
diff --git a/core/system/__init__.py b/core/system/__init__.py
index b768892..c2473b5 100644
--- a/core/system/__init__.py
+++ b/core/system/__init__.py
@@ -3,12 +3,12 @@
 from .encoder import *
 from .vae import VAESystem
 from .ae_ddpm import AE_DDPM
-from .explore import Explore
+# from .explore import Explore
 
 systems = {
     'encoder': EncoderSystem,
     'ddpm': DDPM,
     'vae': VAESystem,
     'ae_ddpm': AE_DDPM,
-    'explore': Explore,
+    # 'explore': Explore,
 }
\ No newline at end of file