diff --git a/train_evaluate/Jenkinsfile_train b/train_evaluate/Jenkinsfile_train
index 5c05c27..39da288 100644
--- a/train_evaluate/Jenkinsfile_train
+++ b/train_evaluate/Jenkinsfile_train
@@ -1,6 +1,6 @@
 pipeline {
    agent {
-      dockerfile true
+      any;
    }
    parameters{
       buildSelector(
@@ -26,13 +26,19 @@ pipeline {
                copyArtifacts fingerprintArtifacts: true, projectName: 's434695-create-dataset', selector: buildParameter('WHICH_BUILD')
             }
         }
-      stage('Docker'){
-         steps{
-            sh 'python3 "./train.py"'
-            sh 'python3 "./sacred1.py"'
-            sh 'python3 "./sacred2.py"'
-         }
-      }
+        stage('docker-training') {
+			steps {
+                script {
+                    def img = docker.build('shroomy/ium:5')
+                    img.inside {
+                        sh "python3 train.py"
+                        sh "python3 sacred1.py"
+                        sh "python3 sacred2.py"
+                 
+                    }
+                }
+			}
+		}
       stage('archiveArtifacts') {
 			steps{
                 archiveArtifacts 'ium_s434695/**'