diff --git a/Jenkinsfile b/Jenkinsfile
index 66a2933..7c42287 100644
--- a/Jenkinsfile
+++ b/Jenkinsfile
@@ -1,6 +1,8 @@
 pipeline {
     agent {
-        dockerfile true
+         dockerfile {
+            additionalBuildArgs '-t ium'
+        }
     }
     parameters {
         string (
@@ -31,8 +33,7 @@ pipeline {
                 withEnv(["KAGGLE_USERNAME=${params.KAGGLE_USERNAME}",
                          "KAGGLE_KEY=${params.KAGGLE_KEY}",
                          "CUTOFF=${params.CUTOFF}"]) {
-                    sh "ls"
-                    sh "cd app"
+                    sh "docker run ium"
                     sh "python3 main_docker.py"
                     archiveArtifacts artifacts: "heart_2020_test.csv, heart_2020_train.csv, heart_2020_sorted.csv", onlyIfSuccessful: true
                 }