diff --git a/Dockerfile b/Dockerfile
index 139b85f..c4ebab4 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -7,6 +7,7 @@ RUN apt install -y unzip
 RUN pip3 install pandas
 RUN pip3 install kaggle
 RUN pip3 install tensorflow
+RUN pip3 install sklearn
 COPY ./data_train ./
 COPY ./data_dev ./
 COPY ./neural_network.sh ./
diff --git a/JenkinsfileNeural b/JenkinsfileNeural
index 810360b..ff6a928 100644
--- a/JenkinsfileNeural
+++ b/JenkinsfileNeural
@@ -12,7 +12,7 @@ node {
 
         }
         stage('Clone repo') {
-                    docker.image("karopa/ium:06").inside {
+                    docker.image("karopa/ium:07").inside {
                         stage('Test') {
                             checkout([$class: 'GitSCM', branches: [[name: '*/master']], doGenerateSubmoduleConfigurations: false, extensions: [], submoduleCfg: [], userRemoteConfigs: [[url: 'https://git.wmi.amu.edu.pl/s434765/ium_434765']]])
                             copyArtifacts fingerprintArtifacts: true, projectName: 's434765-create-dataset', selector: buildParameter("BUILD_SELECTOR")
diff --git a/neural_network.py b/neural_network.py
index 8900780..d9cef78 100644
--- a/neural_network.py
+++ b/neural_network.py
@@ -1,6 +1,5 @@
 import pandas as pd
 import numpy as np
-from keras import optimizers
 from sklearn.metrics import mean_squared_error
 
 from tensorflow import keras