diff --git a/Jenkinsfile b/Jenkinsfile
index c3ec310..c8c3d0f 100644
--- a/Jenkinsfile
+++ b/Jenkinsfile
@@ -20,11 +20,13 @@ pipeline {
             steps {
                 script {
                     def datasetDir = 'football-semantic-segmentation'
-                    sh "mkdir ${datasetDir}"
+                    if (!fileExists(datasetDir)) {
+                        sh "mkdir ${datasetDir}"
+                    }
                     
-                    // Download the dataset
-                    withCredentials([string(value: params.KAGGLE_KEY, variable: 'KAGGLE_API_TOKEN')]) {
-                        sh "KAGGLE_CONFIG_DIR=. kaggle datasets download sadhliroomyprime/football-semantic-segmentation -p ${datasetDir}"
+                    withEnv(["KAGGLE_USERNAME=${params.KAGGLE_USERNAME}",
+                             "KAGGLE_KEY=${params.KAGGLE_KEY}"]) {
+                        sh "kaggle datasets download sadhliroomyprime/football-semantic-segmentation -p ${datasetDir}"
                     }
                 }
             }
@@ -36,7 +38,7 @@ pipeline {
         }
         stage('Checkout') {
             steps {
-                git url: 'https://git.wmi.amu.edu.pl/s495715/iumKC.git'
+                git branch: 'main', url: 'https://git.wmi.amu.edu.pl/s495715/iumKC.git'
             }
         }
     }