diff --git a/Jenkinsfile b/Jenkinsfile
index 7ab7ef3..2d129ab 100644
--- a/Jenkinsfile
+++ b/Jenkinsfile
@@ -28,11 +28,11 @@ pipeline {
                 }
             }
         }
-        stage('Archive dataset and run training') {
-            steps {
-               archiveArtifacts artifacts: 'data/*', onlyIfSuccessful: true
-               build job: 's444409-training/main/'
-            }
+    }
+    post {
+        always {
+            archiveArtifacts artifacts: 'data/*', 
+            onlyIfSuccessful: true
         }
     }
 }
diff --git a/Jenkinsfile-docker b/Jenkinsfile-docker
index 4145331..17900b6 100644
--- a/Jenkinsfile-docker
+++ b/Jenkinsfile-docker
@@ -20,22 +20,16 @@ pipeline {
     }
     
     stages {
-        stage('Get arifacts') {
+        stage('Run data stats') {
             steps {
-                copyArtifacts fingerprintArtifacts: true, projectName: 's444409-create-dataset', selector: lastSuccessful()
-            }
-        }
-
-        stage('Train model') {
-            steps {
-                sh "python train_model.py"
+                sh "python power_plant_data_stats.py"
             }
         }
     }
 
     post {
         always {
-            archiveArtifacts artifacts: 'model_out', 
+            archiveArtifacts artifacts: 'data/*', 
             onlyIfSuccessful: true
         }
     }