diff --git a/cli/train.py b/enhancer/cli/train.py
similarity index 100%
rename from cli/train.py
rename to enhancer/cli/train.py
diff --git a/cli/train_config/config.yaml b/enhancer/cli/train_config/config.yaml
similarity index 100%
rename from cli/train_config/config.yaml
rename to enhancer/cli/train_config/config.yaml
diff --git a/cli/train_config/dataset/DNS-2020.yaml b/enhancer/cli/train_config/dataset/DNS-2020.yaml
similarity index 100%
rename from cli/train_config/dataset/DNS-2020.yaml
rename to enhancer/cli/train_config/dataset/DNS-2020.yaml
diff --git a/cli/train_config/dataset/Vctk.yaml b/enhancer/cli/train_config/dataset/Vctk.yaml
similarity index 100%
rename from cli/train_config/dataset/Vctk.yaml
rename to enhancer/cli/train_config/dataset/Vctk.yaml
diff --git a/enhancer/cli/train_config/dataset/Vctk_local.yaml b/enhancer/cli/train_config/dataset/Vctk_local.yaml
new file mode 100644
index 0000000..b792b71
--- /dev/null
+++ b/enhancer/cli/train_config/dataset/Vctk_local.yaml
@@ -0,0 +1,13 @@
+_target_: enhancer.data.dataset.EnhancerDataset
+name : vctk
+root_dir : /Users/shahules/Myprojects/enhancer/datasets/vctk
+duration : 1.0
+sampling_rate: 16000
+batch_size: 64
+num_workers : 0
+
+files:
+  train_clean : clean_testset_wav
+  test_clean : clean_testset_wav
+  train_noisy : noisy_testset_wav
+  test_noisy : noisy_testset_wav
\ No newline at end of file
diff --git a/cli/train_config/hyperparameters/default.yaml b/enhancer/cli/train_config/hyperparameters/default.yaml
similarity index 100%
rename from cli/train_config/hyperparameters/default.yaml
rename to enhancer/cli/train_config/hyperparameters/default.yaml
diff --git a/cli/train_config/mlflow/experiment.yaml b/enhancer/cli/train_config/mlflow/experiment.yaml
similarity index 100%
rename from cli/train_config/mlflow/experiment.yaml
rename to enhancer/cli/train_config/mlflow/experiment.yaml
diff --git a/cli/train_config/model/Demucs.yaml b/enhancer/cli/train_config/model/Demucs.yaml
similarity index 100%
rename from cli/train_config/model/Demucs.yaml
rename to enhancer/cli/train_config/model/Demucs.yaml
diff --git a/cli/train_config/model/WaveUnet.yaml b/enhancer/cli/train_config/model/WaveUnet.yaml
similarity index 100%
rename from cli/train_config/model/WaveUnet.yaml
rename to enhancer/cli/train_config/model/WaveUnet.yaml
diff --git a/cli/train_config/optimizer/Adam.yaml b/enhancer/cli/train_config/optimizer/Adam.yaml
similarity index 100%
rename from cli/train_config/optimizer/Adam.yaml
rename to enhancer/cli/train_config/optimizer/Adam.yaml
diff --git a/cli/train_config/trainer/default.yaml b/enhancer/cli/train_config/trainer/default.yaml
similarity index 100%
rename from cli/train_config/trainer/default.yaml
rename to enhancer/cli/train_config/trainer/default.yaml
diff --git a/cli/train_config/trainer/fastrun_dev.yaml b/enhancer/cli/train_config/trainer/fastrun_dev.yaml
similarity index 100%
rename from cli/train_config/trainer/fastrun_dev.yaml
rename to enhancer/cli/train_config/trainer/fastrun_dev.yaml