diff --git a/Dockerfile b/Dockerfile
index 43d97b0fe07ce25c9e3c4976eaae761b0bf60e61..48c9bce5424a3a988a0885f66abca90a1b676a58 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -2,15 +2,26 @@ FROM quay.io/jupyter/tensorflow-notebook:hub-5.2.0
 
 USER root
 
+# Comment: force full rebuild
+
 # Update Pip and conda
 RUN pip3 install --upgrade pip
 RUN mamba update mamba
 
 # Conda packages
-RUN mamba install -c conda-forge jupyter-server-proxy==4.4.0
+RUN mamba install -c conda-forge jupyter-server-proxy=4.4.0
+RUN mamba install -c conda-forge jupyter_contrib_nbextensions
+RUN mamba install -c conda-forge myst-nb
+RUN mamba install -c conda-forge ipympl
+RUN mamba install -c conda-forge ipywidgets
+
 RUN mamba install -c conda-forge control
 RUN mamba install -c conda-forge casadi
-RUN mamba install -c conda-forge jax
 RUN mamba install -c conda-forge networkx
+RUN mamba install -c conda-forge pillow
+RUN mamba install -c conda-forge jaxlib
+RUN mamba install -c conda-forge jax
+RUN mamba install -c conda-forge flax
 
-# Install pip packages
\ No newline at end of file
+# Install pip packages
+#RUN pip install rlax
\ No newline at end of file