diff --git a/Dockerfile b/Dockerfile
index 8d205c1..4928c9a 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -2,16 +2,11 @@ FROM ubuntu:latest
 
 WORKDIR /ium
 
-RUN apt update && apt install -y python3-pip unzip
+RUN apt update && apt install -y python3-pip
 
-RUN pip3 install --user kaggle
-RUN echo "export PATH=\"\$HOME/.local/bin:\$PATH\"" >> ~/.bashrc
+RUN pip3 install pandas
+RUN pip3 install numpy
+RUN pip3 install sklearn
 
-RUN pip3 install --user pandas
-RUN pip3 install --user numpy
-RUN pip3 install --user sklearn
-
-COPY ./download_dataset.sh ./
-COPY ./process_data.py ./
-
-RUN chmod u+x ./download_dataset.sh ./process_data.py
\ No newline at end of file
+COPY ./lego_sets.csv ./
+COPY ./process_dataset.py ./
\ No newline at end of file
diff --git a/Jenkinsfile b/Jenkinsfile
index 82f4c17..7897f1b 100644
--- a/Jenkinsfile
+++ b/Jenkinsfile
@@ -13,15 +13,13 @@ pipeline {
 		)
 	}
     agent {
-        dockerfile {
-            additionalBuildArgs "--build-arg KAGGLE_USERNAME=${params.KAGGLE_USERNAME} --build-arg KAGGLE_KEY=${params.KAGGLE_KEY}"
+        dockerfile true
         }
     }
     stages {
         stage('Stage 1') {
             steps {
-					echo 'Downloading dataset...' 
-					sh './download_dataset.sh'
+				    sh 'chmod u+x ./process_dataset.py'
 					echo 'Dataset downloaded'
                     echo 'Processing dataset...'
                     sh 'python3 process_dataset.py'