diff --git a/Jenkinsfile b/Jenkinsfile
index a5da84d..06469a0 100644
--- a/Jenkinsfile
+++ b/Jenkinsfile
@@ -39,6 +39,12 @@ pipeline {
        }
     }
   stage("Data Preparation") {
+            agent {
+               		dockerfile {
+			            additionalBuildArgs '-t ium'
+                    	args '-e KAGGLE_USERNAME=${params.KAGGLE_USERNAME} -e KAGGLE_KEY=${params.KAGGLE_KEY}'
+		        }
+            }
             steps {
                 sh "python create_data.py"
 				archiveArtifacts 'KaggleV2-May-2016.csv'