diff --git a/docker/Dockerfile.jupyter b/docker/Dockerfile.jupyter
index 40c5a63..62ac7b3 100644
--- a/docker/Dockerfile.jupyter
+++ b/docker/Dockerfile.jupyter
@@ -164,6 +164,20 @@ RUN pip install aeon
 #     && conda deactivate
 # # ---- Zero shot ---- #
 
+COPY --chown=$UID:$GID ./Makefile $HOME/work/
+COPY --chown=$UID:$GID ./nbs $HOME/work/
+COPY --chown=$UID:$GID ./settings.ini $HOME/work/
+COPY --chown=$UID:$GID ./dvats $HOME/work/
+COPY --chown=$UID:$GID ./LICENSE $HOME/work/
+COPY --chown=$UID:$GID ./MANIFEST.in $HOME/work/
+COPY --chown=$UID:$GID ./nbs_pipeline $HOME/work/
+COPY --chown=$UID:$GID ./setup.py $HOME/work/
+COPY --chown=$UID:$GID ./README.md $HOME/work/
+
+RUN cd $HOME/work  \
+    && pip install -e $HOME/work
+
+
 ARG JUPYTER_TOKEN
 ENV JUPYTER_TOKEN=${JUPYTER_TOKEN}
 
diff --git a/docker/entrypoint.sh b/docker/entrypoint.sh
index e5ac5b3..51e2ece 100644
--- a/docker/entrypoint.sh
+++ b/docker/entrypoint.sh
@@ -2,8 +2,8 @@
 set -e
 #echo $ENV_PREFIX
 #conda list 
-ls -la /home/$USER/work
-pip install -e /home/$USER/work
+#ls -la /home/$USER/work
+#pip install -e /home/$USER/work
 
 echo $WANDB_ENTITY $USER $WANDB_PROJECT