diff --git a/.dvc/config b/.dvc/config
index e69de29..c02d6a2 100644
--- a/.dvc/config
+++ b/.dvc/config
@@ -0,0 +1,4 @@
+[core]
+    remote = ium_ssh_remote
+['remote "ium_ssh_remote"']
+    url = ssh://ium-sftp@tzietkiewicz.vm.wmi.amu.edu.pl/ium-sftp
diff --git a/MoviesOnStreamingPlatforms_updated.dev.dvc b/MoviesOnStreamingPlatforms_updated.dev.dvc
new file mode 100644
index 0000000..c595c55
--- /dev/null
+++ b/MoviesOnStreamingPlatforms_updated.dev.dvc
@@ -0,0 +1,4 @@
+outs:
+- md5: bfca23da2c854d8e5713b4bca178f647
+  size: 53794
+  path: MoviesOnStreamingPlatforms_updated.dev
diff --git a/MoviesOnStreamingPlatforms_updated.test.dvc b/MoviesOnStreamingPlatforms_updated.test.dvc
new file mode 100644
index 0000000..3e0e588
--- /dev/null
+++ b/MoviesOnStreamingPlatforms_updated.test.dvc
@@ -0,0 +1,4 @@
+outs:
+- md5: 87297b0971adc0eff135210d66e01af7
+  size: 54419
+  path: MoviesOnStreamingPlatforms_updated.test
diff --git a/MoviesOnStreamingPlatforms_updated.train.dvc b/MoviesOnStreamingPlatforms_updated.train.dvc
new file mode 100644
index 0000000..a11cc50
--- /dev/null
+++ b/MoviesOnStreamingPlatforms_updated.train.dvc
@@ -0,0 +1,4 @@
+outs:
+- md5: 4c81cf7c67631cdee0fb34238ebe35a1
+  size: 429007
+  path: MoviesOnStreamingPlatforms_updated.train
diff --git a/dvc.yaml b/dvc.yaml
new file mode 100644
index 0000000..186335f
--- /dev/null
+++ b/dvc.yaml
@@ -0,0 +1,16 @@
+stages:
+    preprocessing:
+        cmd: python3 script.py
+        deps:
+          - script.py
+        outs:
+          - MoviesOnStreamingPlatforms_updated.train
+          - MoviesOnStreamingPlatforms_updated.test
+          - MoviesOnStreamingPlatforms_updated.dev
+    training:
+        cmd: python3 training.py
+        deps:
+          - training.py
+          - MoviesOnStreamingPlatforms_updated.train
+        outs:
+          - linear_regression.h5
\ No newline at end of file