diff --git a/requirements.txt b/requirements.txt
index 4a02db0..0eaaff2 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,13 +1,12 @@
 h5py
-lightning
+pytorch-lightning
 matplotlib
 numpy
 PyYAML
-PyYAML
-scikit_learn
+scikit-learn
 timm
 torch
 onnx
 onnxruntime
---index-url https://riahub.qoherent.internal:3000/api/packages/qoherent/pypi/simple/
+--index-url https://git.riahub.ai/api/packages/qoherent/pypi/simple/ 
 utils
diff --git a/scripts/dataset_building/data_gen.py b/scripts/dataset_building/data_gen.py
index 27d09b7..7380f97 100644
--- a/scripts/dataset_building/data_gen.py
+++ b/scripts/dataset_building/data_gen.py
@@ -75,3 +75,4 @@ if __name__ == "__main__":
     )
     args = p.parse_args()
     generate_modulated_signals(args.output_dir)
+    print("generated data to " + args.output_dirˍ)