diff --git a/dvc.yaml b/dvc.yaml
index 3e57527..56df93e 100644
--- a/dvc.yaml
+++ b/dvc.yaml
@@ -1,5 +1,11 @@
 stages:
   prepare:
-    cmd: '" -d" ml_prepare.py -o training_data.csv test_data.csv " python" ml_prepare.py'
+    cmd: python3 ml_prepare.py
+    deps:
+      - training_data.csv
+      - test_data.csv
+      - ml_prepare.py
   training:
-    cmd: '" -d" ml_training.py " -o" trained_model/ " python" ml_training.py 15'
+    cmd: python3 ml_training.py 15
+    deps:
+      - ml_training.py