diff --git a/Dockerfile b/Dockerfile
index 52f5f11..46cd70a 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -2,17 +2,16 @@ FROM ubuntu
 
 RUN apt-get update && apt-get install -y python3 python3-pip unzip
 
-RUN python3 -m pip install pandas numpy tensorflow imbalanced-learn sklearn sacred
+RUN python3 -m pip install pandas numpy tensorflow imbalanced-learn sklearn sacred pymongo
 RUN apt-get install -y git
 
 COPY train.py /app/train.py
 COPY predictions.py /app/predictions.py
 COPY data.csv /app/data.csv
-# COPY test_data.csv /app/test_data.csv
 
 WORKDIR /app
 
-export SACRED_IGNORE_GIT=TRUE
+RUN export SACRED_IGNORE_GIT=TRUE
 RUN python3 train.py
 
-RUN python3 predictions.py
\ No newline at end of file
+CMD ["python3", "predictions.py"]
diff --git a/JenkinsfileDL b/JenkinsfileDL
index f8e83d6..6647c29 100644
--- a/JenkinsfileDL
+++ b/JenkinsfileDL
@@ -3,6 +3,9 @@ pipeline {
         docker {
             image 'python:3.11'
             args '-v /root/.cache:/root/.cache -u root'
+            environment {
+                SACRED_IGNORE_GIT = 'TRUE'
+            }
         }
     }
     parameters {
@@ -24,8 +27,7 @@ pipeline {
         stage('Trenuj model') {
             steps {
                 script {
-                    // sh "python3 train.py --epochs $EPOCHS"
-                    sh "python3 train.py"
+                    sh "python3 train.py --epochs $EPOCHS"
                 }
             }
         }