diff --git a/dvc.yaml b/dvc.yaml
index 649c7df..1616783 100644
--- a/dvc.yaml
+++ b/dvc.yaml
@@ -3,6 +3,16 @@ stages:
     cmd: python3 process_dataset.py
     deps:
     - lego_sets.csv
+    outs:
+    - lego_sets_clean.csv
+    - lego_sets_clean_train.csv
+    - lego_sets_clean_valid.csv
+    - lego_sets_clean_test.csv
   train:
     cmd: python3 simple_regression.py 100
-  
+    deps:
+    - lego_sets_clean_train.csv
+    - lego_sets_clean_test.csv
+    outs:
+    - lego_reg_model
+    - lego_reg_results.csv
\ No newline at end of file