diff --git a/Dockerfile b/Dockerfile
index 7a0d156..ab2326e 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -9,4 +9,4 @@ RUN pip3 install numpy
 
 WORKDIR /app
 
-#COPY ./training.py ./
+COPY ./training.py ./
diff --git a/training.Jenkinsfile b/training.Jenkinsfile
index 1efcbbc..06bbf0c 100644
--- a/training.Jenkinsfile
+++ b/training.Jenkinsfile
@@ -23,6 +23,7 @@ pipeline {
         stage('Script') {
             steps {
             	copyArtifacts filter: '*', projectName:'s444421-create-dataset', selector: buildParameter('BUILD_SELECTOR')
+            	sh 'chmod u+x ./training.py'
             	sh 'ipython ./training.py $EPOCHS'
             	archiveArtifacts artifacts: 'model.pth'
             }
diff --git a/training.py b/training.py
old mode 100644
new mode 100755