diff --git a/Jenkinsfile b/Jenkinsfile
index 6aebd13..f76fe4b 100644
--- a/Jenkinsfile
+++ b/Jenkinsfile
@@ -39,6 +39,9 @@ pipeline {
             }
         }
   stage("Dara Preparation") {
+      agent {
+                docker { image 'ium_478855' }
+            }
             steps {
                 sh 'echo "PREPARATION"'
 				sh "chmod u+x ./preparation.sh"
diff --git a/create_data.py b/create_data.py
index 30bc8ae..3f9918e 100644
--- a/create_data.py
+++ b/create_data.py
@@ -22,4 +22,6 @@ no_shows["Age"]=(no_shows["Age"]-no_shows["Age"].min())/(no_shows["Age"].max()-n
 X = no_shows.drop(columns=['No-show'])
 y = no_shows['No-show']
 
-X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
\ No newline at end of file
+X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
+
+print("Quiting create_data.py")
\ No newline at end of file
diff --git a/preparation.sh b/preparation.sh
index dfe1747..a086870 100644
--- a/preparation.sh
+++ b/preparation.sh
@@ -1 +1,2 @@
-echo "Preparation inner"
\ No newline at end of file
+echo "Preparation inner"
+python create_data.py
\ No newline at end of file