diff --git a/.riahub/workflows/workflow.yaml b/.riahub/workflows/workflow.yaml
index 65e7aee..eabfbe7 100644
--- a/.riahub/workflows/workflow.yaml
+++ b/.riahub/workflows/workflow.yaml
@@ -40,7 +40,7 @@ jobs:
       - name: 1. Build HDF5 Dataset
         run: |
          mkdir -p data
-         PYTHONPATH=. python scripts/produce_dataset.py
+         PYTHONPATH=. python data/scripts/produce_dataset.py
          echo "datasets produced successfully"
         shell: bash
 
diff --git a/conf/app.yaml b/conf/app.yaml
index 9db0416..e8e3d3d 100644
--- a/conf/app.yaml
+++ b/conf/app.yaml
@@ -3,24 +3,25 @@ general:
   run_mode: prod
 
 dataset:
-  input_dir: "recordings"
+  input_dir: data/recordings
   num_slices: 8
   train_split: 0.8
   val_split : 0.2
-  seed: 42
-  output_dir: "data"
+  seed: 25
+  modulation_types: [bpsk, qpsk, qam16, qam64]
+  output_dir: data/dataset
 
 training:
   batch_size: 64
   epochs: 50
   learning_rate: 0.001
-  checkpoint_path: "checkpoints/model.ckpt"
+  checkpoint_path: checkpoints/model.ckpt
   use_gpu: true
 
 inference:
-  model_path: "checkpoints/model.ckpt"
+  model_path: checkpoints/model.ckpt
   num_classes: 10
-  output_path: "results/output.json"
+  output_path: results/output.json
 
 app:
-  build_dir: "dist"
\ No newline at end of file
+  build_dir: dist
\ No newline at end of file
diff --git a/helpers/app_settings.py b/helpers/app_settings.py
index a2cc422..2b2c861 100644
--- a/helpers/app_settings.py
+++ b/helpers/app_settings.py
@@ -14,6 +14,7 @@ class DataSetConfig:
     num_slices: int
     train_split: float
     seed: int
+    modulation_types: list
     val_split: float
     output_dir: str