diff --git a/Jenkinsfile b/Jenkinsfile
index ace8b03..43507dc 100644
--- a/Jenkinsfile
+++ b/Jenkinsfile
@@ -1,5 +1,7 @@
 pipeline {
-    agent any
+    agent {
+        dockerfile true
+    }
 
     parameters {
         password (
@@ -29,9 +31,9 @@ pipeline {
         stage('Download dataset') {
             steps {
                 withEnv(["KAGGLE_USERNAME=${params.KAGGLE_USERNAME}", "KAGGLE_KEY=${params.KAGGLE_KEY}", "CUTOFF=${params.CUTOFF}"]) {
-                    sh "chmod +x download_dataset.sh"
-                    sh "./download_dataset.sh $CUTOFF"
-                    archiveArtifacts artifacts: 'data.csv,train.csv,dev.csv,test.csv', onlyIfSuccessful: true
+                    sh "chmod +x ./download_dataset.py"
+                    sh "python3 ./download_dataset.py $CUTOFF"
+                    archiveArtifacts artifacts: './datasets/data.csv,./datasets/train.csv,./datasets/dev.csv,./datasets/test.csv', onlyIfSuccessful: true
                 }
             }
         }
diff --git a/statistics/Jenkinsfile b/statistics/Jenkinsfile
index 6c92eec..ba47ed1 100644
--- a/statistics/Jenkinsfile
+++ b/statistics/Jenkinsfile
@@ -1,5 +1,9 @@
 pipeline {
-    agent any
+    agent {
+        docker {
+            image 'dzikafoczka/ium:latest'
+        }
+    }
 
     parameters {
         buildSelector(
@@ -24,8 +28,8 @@ pipeline {
 
         stage('Get statistics') {
             steps {
-                sh "chmod +x get_stats.sh"
-                sh "./get_stats.sh"
+                sh "chmod +x get_stats.py"
+                sh "python3 get_stats.py"
                 archiveArtifacts artifacts: 'stats.txt', onlyIfSuccessful: true
             }
         }