diff --git a/Jenkinsfile-create-dataset-docker b/Jenkinsfile-create-dataset-docker
index d28e69e..07c8c4b 100644
--- a/Jenkinsfile-create-dataset-docker
+++ b/Jenkinsfile-create-dataset-docker
@@ -1,7 +1,5 @@
 pipeline {
-    agent {
-        dockerfile true
-    }
+    agent any
     parameters{
      string(
          defaultValue: 'piotrwrzodak',
@@ -38,10 +36,21 @@ pipeline {
                    sh 'kaggle datasets download -d thedevastator/airbnb-prices-in-european-cities'
                    sh 'unzip airbnb-prices-in-european-cities.zip -d data'
                    sh 'ls'
-                   sh 'python create-dataset.py'
-                   archiveArtifacts artifacts: 'data/barcelona_weekends.train.csv, data/barcelona_weekends.dev.csv, data/barcelona_weekends.test.csv', fingerprint: true
                 }
             }
         }
+        stage('Docker') {
+            agent {
+                dockerfile true
+            }
+            steps {
+                sh 'python create-dataset.py'
+            }
+        }
+        stage('Archive') {
+            steps {
+                archiveArtifacts artifacts: 'data/barcelona_weekends.train.csv, data/barcelona_weekends.dev.csv, data/barcelona_weekends.test.csv', fingerprint: true
+            }
+        }
    }
 }
\ No newline at end of file