diff --git a/dvc.yaml b/dvc.yaml
index 2c85c35..fbf0d84 100644
--- a/dvc.yaml
+++ b/dvc.yaml
@@ -1,11 +1,10 @@
 stages:
   prepare_datasets:
-    cmd: python3 prepare_datasets.py X_train.csv X_test.csv y_train.csv y_test.csv
+    cmd: python3 prepare_datasets.py 
     deps:
     - data/bodyPerformance.csv
     - prepare_datasets.py
-    outs:
-    - X_test.csv
-    - X_train.csv
-    - y_test.csv
-    - y_train.csv
+  train:
+    cmd: python train.py
+    deps:
+    - train.py