diff --git a/Jenkinsfile_train b/Jenkinsfile_train
index af4c81d..2a35e40 100644
--- a/Jenkinsfile_train
+++ b/Jenkinsfile_train
@@ -17,16 +17,31 @@ pipeline {
             description: 'Kaggle token taken from kaggle.json file, as described in https://github.com/Kaggle/kaggle-api#api-credentials',
             name: 'KAGGLE_KEY'
 		)
+		
+        string(
+            defaultValue: '1000',
+            description: 'Number of epochs',
+            name: 'EPOCHS',
+            trim: false
      }
 
    stages {
 	stage('First'){
 		steps {
-                        copyArtifacts filter: '*', projectName: 's478815-create-dataset'
-			sh 'python3 ./DL.py'
+            copyArtifacts filter: '*', projectName: 's478815-create-dataset'
+			sh 'python3 ./DL.py $EPOCHS'
 			archiveArtifacts artifacts: 'modelP.pkl', followSymlinks: false
 			
 		}
 	}
     }
+
+	post {
+        success {
+            emailext body: 'SUCCESS', subject: 's478815-training', to: 'e19191c5.uam.onmicrosoft.com@emea.teams.ms'
+        }
+        failure {
+            emailext body: 'FAILURE', subject: 's478815-training', to: 'e19191c5.uam.onmicrosoft.com@emea.teams.ms'
+        }
+	}
 }
\ No newline at end of file