diff --git a/Dockerfile b/Dockerfile
index 6be5675..631ad22 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -2,17 +2,13 @@ FROM kubeflownotebookswg/jupyter-pytorch-cuda-full:v1.7.0
 
 USER root
 
-RUN apt-get update -y
-RUN apt-get upgrade -y
+RUN username=jovyan \
+  password=jovyan \
+  adduser --gecos "" --disabled-password $username \
+  chpasswd <<<"$username:$password"
 
-USER jovyan
-
-RUN pip install plotly
-RUN pip install darts
-RUN pip install -U torch torchaudio --no-cache-dir
-RUN pip install wandb
-RUN pip install ipywidgets==8.0.4
-RUN jupyter nbextension enable --py widgetsnbextension
+# Allow using 'su -' to make sandbox changes
+RUN chpasswd <<<"root:root" && \
+    sed -Ei 's/(.*pam_deny.so)/# \1/' /etc/pam.d/su
 
-RUN echo "root:1234patata" | chpasswd
-RUN echo "jovyan:1234patata" | chpasswd
+USER jovyan