diff --git a/Jenkinsfile b/Jenkinsfile
index 02787ce..728eeac 100644
--- a/Jenkinsfile
+++ b/Jenkinsfile
@@ -15,17 +15,18 @@ pipeline {
         }
 
         stage('Download, Process, and Split Dataset') {
+            agent { dockerfile {
+                    args '--build-arg="KAGGLE_KEY=${env.KAGGLE_KEY}" --build-arg="KAGGLE_USERNAME=${env.KAGGLE_USERNAME}"'
+                }
+             }
+            steps {
                 withEnv([
                     "KAGGLE_USERNAME=${env.KAGGLE_USERNAME}",
                     "KAGGLE_KEY=${env.KAGGLE_KEY}"
                 ]) {
-                    agent { dockerfile true}
-                    steps {
-
-                            sh "python3 IUM_02.py"
-
-                    }
-              }
+                    sh "python3 IUM_02.py"
+                }
+            }
         }
         stage('Archive Results') {
             steps {