diff --git a/JenkinsfileNeural b/JenkinsfileNeural
index 64929ca..b48c042 100644
--- a/JenkinsfileNeural
+++ b/JenkinsfileNeural
@@ -1,6 +1,5 @@
 node {
-          try {
-          stage('Preparation') {
+         stage('Preparation') {
 
                 properties([
                     parameters([
@@ -17,30 +16,29 @@ node {
 
         }
         stage('Clone repo') {
+                    try { docker.image("karopa/ium:11").inside {
                         stage('Test') {
-                            docker.image("karopa/ium:11").inside {
-
-                                    checkout([$class: 'GitSCM', branches: [[name: '*/master']], doGenerateSubmoduleConfigurations: false, extensions: [], submoduleCfg: [], userRemoteConfigs: [[url: 'https://git.wmi.amu.edu.pl/s434765/ium_434765']]])
-                                    copyArtifacts fingerprintArtifacts: true, projectName: 's434765-create-dataset', selector: buildParameter("BUILD_SELECTOR")
-                                    sh '''
-                                        #!/usr/bin/env bash
-                                        chmod 777 neural_network.sh ${params.KAGGLE_USERNAME}
-                                        ./neural_network.sh | tee output.txt
-                                    '''
-                                    archiveArtifacts 'output.txt'
-                                    archiveArtifacts 'model/**/*.*'
-                                    emailext body: 's434765',
-                                    subject: "Successful build",
-                                    to: "26ab8f35.uam.onmicrosoft.com@emea.teams.ms"
-
-                            }
+                            checkout([$class: 'GitSCM', branches: [[name: '*/master']], doGenerateSubmoduleConfigurations: false, extensions: [], submoduleCfg: [], userRemoteConfigs: [[url: 'https://git.wmi.amu.edu.pl/s434765/ium_434765']]])
+                            copyArtifacts fingerprintArtifacts: true, projectName: 's434765-create-dataset', selector: buildParameter("BUILD_SELECTOR")
+                            sh '''
+                                #!/usr/bin/env bash
+                                chmod 777 neural_network.sh ${params.EPOCHS}
+                                ./neural_network.sh | tee output.txt
+                            '''
+                            archiveArtifacts 'output.txt'
+                            archiveArtifacts 'model/**/*.*'
                         }
-        }
-        }
-            catch (e) {
-               emailext body: 's434765',
-               subject: "Failed build",
-               to: "26ab8f35.uam.onmicrosoft.com@emea.teams.ms"
-               throw e
+                        emailext body: 'Successful build',
+                        subject: "Successful build",
+                        to: "26ab8f35.uam.onmicrosoft.com@emea.teams.ms"
+                        }
+
+                    }
+                     catch (e) {
+                            emailext body: 'Failed build',
+                            subject: "Failed build",
+                            to: "26ab8f35.uam.onmicrosoft.com@emea.teams.ms"
+                            throw e
+                     }
             }
-}
+        }