diff --git a/Jenkinsfile b/Jenkinsfile
index 869bd1e..e84df64 100644
--- a/Jenkinsfile
+++ b/Jenkinsfile
@@ -1,11 +1,9 @@
 pipeline {
     agent { dockerfile true }
     stages {
-        stage('Dataset download and stats') {
+        stage('Model training') {
             steps {
-                sh "export KAGGLE_CONFIG_DIR='${env.WORKSPACE}/kaggle_config/'"
-                sh "chmod +x -R ${env.WORKSPACE}"
-                sh './dataset_download_and_run.sh'
+                sh './train.sh'
             }
         }
 
@@ -15,7 +13,6 @@ pipeline {
             archiveArtifacts 'test.csv'
             archiveArtifacts 'train.csv'
             archiveArtifacts 'valid.csv'
-            archiveArtifacts 'extracted_dataset/2022/processed.csv'
          }
       }
     }