diff --git a/Dockerfile b/Dockerfile
index b3839ea..0419151 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -2,7 +2,7 @@ FROM ubuntu:latest
 
 RUN apt-get update && apt-get install -y python3-pip && pip3 install setuptools && pip3 install numpy && pip3 install pandas && pip3 install wget && pip3 install scikit-learn && pip3 install matplotlib && rm -rf /var/lib/apt/lists/*
 RUN pip3 install torch torchvision torchaudio
-RUN pip3 install sacred && pip3 install GitPython
+RUN pip3 install sacred && pip3 install GitPython && pip3 install pymongo
 WORKDIR /app
 
 COPY ./create.py ./
diff --git a/pytorch-training-eval/Dockerfile b/pytorch-training-eval/Dockerfile
index 720992d..6f01ac8 100644
--- a/pytorch-training-eval/Dockerfile
+++ b/pytorch-training-eval/Dockerfile
@@ -2,7 +2,7 @@ FROM ubuntu:latest
 
 RUN apt-get update && apt-get install -y python3-pip && pip3 install setuptools && pip3 install numpy && pip3 install pandas && pip3 install wget && pip3 install scikit-learn && pip3 install matplotlib  && rm -rf /var/lib/apt/lists/*
 RUN pip3 install torch torchvision torchaudio 
-RUN pip3 install sacred && pip3 install GitPython
+RUN pip3 install sacred && pip3 install GitPython && pip3 install pymongo
 WORKDIR /app 
 
 COPY ./../create.py ./