diff --git a/Dockerfile b/Dockerfile
index 27369a9..2d3a3d9 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,6 +1,7 @@
 FROM ubuntu:latest
 
 RUN apt update && apt install -y figlet
+RUN apt install unzip -y
 RUN apt install python3-pip -y
 
 RUN pip3 install kaggle
diff --git a/Jenkinsfile_train b/Jenkinsfile_train
index a675b69..731800e 100644
--- a/Jenkinsfile_train
+++ b/Jenkinsfile_train
@@ -10,7 +10,7 @@ node {
      checkout([$class: 'GitSCM', branches: [[name: '*/master']], extensions: [], userRemoteConfigs: [[credentialsId: 's487197', url: 'https://git.wmi.amu.edu.pl/s487197/ium_487197']]])
  }
  stage('Dockerfile'){
-    def testImage = docker.image('s487197/ium:33')
+    def testImage = docker.image('s487197/ium:34')
 	testImage.inside{ 
          copyArtifacts filter: 'baltimore_train.csv', projectName: 's487197-create-dataset'
 		 sh "python3 ium_train.py"