diff --git a/Dockerfile b/Dockerfile
index 2becfba..5515353 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -14,10 +14,11 @@ RUN echo 'export PATH=/opt/conda/bin:$PATH' > /etc/profile.d/conda.sh && \
 ENV PATH /opt/conda/bin:$PATH
 ENV PYTHONPATH /gale:$PYTHONPATH
 
-#Create gale conda environment (like cd gale)
+# Create gale conda environment (like cd gale)
 WORKDIR /gale
 ADD environment.yml .
 RUN conda env create -f environment.yml && conda clean -a -y
+RUN conda init bash
 
 # Add the path of the python interpreter (like source activate gale)
 ENV PATH /opt/conda/envs/gale/bin/:$PATH
diff --git a/environment.yml b/environment.yml
index c3f19b2..ab91e08 100644
--- a/environment.yml
+++ b/environment.yml
@@ -19,6 +19,7 @@ dependencies:
   - gstreamer=1.14.0=hb453b48_1
   - icu=58.2=h9c2bf20_1
   - intel-openmp=2020.0=166
+  - joblib=0.16.0=py_0
   - jpeg=9b=h024ee3a_2
   - kiwisolver=1.1.0=py37he6710b0_0
   - libedit=3.1.20181209=hc058e9b_0
@@ -55,6 +56,7 @@ dependencies:
   - pytz=2019.3=py_0
   - qt=5.9.7=h5867ecd_1
   - readline=7.0=h7b6447c_5
+  - scikit-learn=0.23.1=py37h423224d_0
   - scipy=1.4.1=py37h0b6359f_0
   - seaborn=0.10.0=py_0
   - setuptools=45.2.0=py37_0