diff --git a/dvc.yaml b/dvc.yaml
index a60f204..9d60ad3 100644
--- a/dvc.yaml
+++ b/dvc.yaml
@@ -4,3 +4,8 @@ stages:
     deps:
     - data/bodyPerformance.csv
     - prepare_datasets.py
+    outs:
+    - X_test.csv
+    - X_train.csv
+    - y_test.csv
+    - y_train.csv