diff --git a/docker/Dockerfile b/docker/Dockerfile
index 0c548507..9cc947f3 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -1,4 +1,5 @@
-FROM runpod/base:0.6.2-cuda12.1.0
+FROM runpod/base:0.6.2-cuda12.2.0
+
 LABEL authors="jaret"
 
 # Install dependencies
@@ -17,5 +18,10 @@ RUN python -m pip install -r requirements.txt
 
 RUN apt-get install -y tmux nvtop htop
 
+RUN pip install jupyterlab
+
+# symlink app to workspace
+RUN ln -s /app/ai-toolkit /workspace/ai-toolkit
+
 WORKDIR /
 CMD ["/start.sh"]
\ No newline at end of file