diff --git a/Jenkinsfile b/Jenkinsfile
index 122f65c..b5f757d 100644
--- a/Jenkinsfile
+++ b/Jenkinsfile
@@ -1,8 +1,6 @@
 pipeline {
     agent { 
-        dockerfile {
-            additionalBuildArgs '-t ium'
-        }
+        dockerfile  true
     }
     parameters {
         string(
@@ -31,6 +29,7 @@ pipeline {
                 "KAGGLE_KEY=${params.KAGGLE_KEY}"]) {
                     sh 'echo KAGGLE_USERNAME: $KAGGLE_USERNAME'
                     sh 'python3 ./data_expl.py'
+                    sh 'python3 ./nn_train.py'
                     archiveArtifacts artifacts: 'apps_train.csv, apps_test.csv, apps_validate.csv'
                 }
             }