diff --git a/dvc.lock b/dvc.lock
new file mode 100644
index 0000000..1392acf
--- /dev/null
+++ b/dvc.lock
@@ -0,0 +1,28 @@
+schema: '2.0'
+stages:
+  preprocess:
+    cmd: python preprocesing.py 1000
+    deps:
+    - path: preprocesing.py
+      md5: 5fc702becd41811c92fc85e4ec899d52
+      size: 904
+    - path: who_suicide_statistics.csv
+      md5: ba0fd3d7333524d077791b70048b5067
+      size: 1895346
+    outs:
+    - path: train.csv
+      md5: eefeb0d1ce453b32aea98b458002f8b0
+      size: 164096
+  train:
+    cmd: python training.py 5 10
+    deps:
+    - path: train.csv
+      md5: eefeb0d1ce453b32aea98b458002f8b0
+      size: 164096
+    - path: training.py
+      md5: d08d4fc890592243d017e91335c2e1d0
+      size: 2598
+    outs:
+    - path: suicide_model.h5
+      md5: 9595399d5870d7a0352a7c56c5aaa8f1
+      size: 12496
diff --git a/dvc.yaml b/dvc.yaml
index 48f6552..0fa07bf 100644
--- a/dvc.yaml
+++ b/dvc.yaml
@@ -1,17 +1,16 @@
 stages:
     preprocess:
-      cmd: python3 preprocesing.py
+      cmd: python3 preprocesing.py 1000
       deps:
         - preprocesing.py
         - who_suicide_statistics.csv
       outs:
         - train.csv
     train:
-      cmd: python3 training.py
+      cmd: python3 training.py 5 10
       deps:
       - training.py
       - train.csv
       outs:
-      - results.csv
       - suicide_model.h5