diff --git a/Jenkinsfile b/Jenkinsfile
index fba3759..1267201 100644
--- a/Jenkinsfile
+++ b/Jenkinsfile
@@ -1,4 +1,7 @@
 pipeline {
+	agent {
+		dockerfile true
+	}
     parameters {
         string (
             defaultValue: 'kamilguttmann',
@@ -17,33 +20,25 @@ pipeline {
             name: 'CUTOFF'
         )
     }
-    agent none
     stages {
         stage('Hello') {
-		    agent any
             steps {
                 echo 'Hello World'
             }
         }
         stage("Checkout git") {
-		    agent any
             steps {
 	        checkout([$class: 'GitSCM', branches: [[name: '*/master']], extensions: [], userRemoteConfigs: [[credentialsId: 's444380', url: 'https://git.wmi.amu.edu.pl/s444380/ium_444380.git']]])
             }
         }
         stage("Download data") {
-		    agent {
-				dockerfile {
-					args '-e KAGGLE_USERNAME=${params.KAGGLE_USERNAME} -e KAGGLE_KEY=${params.KAGGLE_KEY} -e CUTOFF=${params.CUTOFF}'
-				}
-			}
             steps {
-                    sh "echo $KAGGLE_USERNAME $CUTOFF"
-					sh "KAGGLE_USERNAME=${params.KAGGLE_USERNAME}"
-					sh "KAGGLE_KEY=${params.KAGGLE_KEY}"
+				withEnv(["KAGGLE_USERNAME=${params.KAGGLE_USERNAME}",
+					"KAGGLE_KEY=${params.KAGGLE_KEY}"]) {
                     sh "./download_data.sh"
-                    sh "python3 clean_and_split_data.py"                    
+                    sh "python3 clean_and_split_data.py"  
                     archiveArtifacts artifacts: "crime.test, crime.dev, crime.train", onlyIfSuccessful: true
+                }
             }
         }
     }