diff --git a/deepspace/Dockerfile b/deepspace/Dockerfile
index 82c5965..16fc77f 100755
--- a/deepspace/Dockerfile
+++ b/deepspace/Dockerfile
@@ -1,21 +1,34 @@
 ARG CUDA_VERSION=12.6.0
-
 FROM nvidia/cuda:${CUDA_VERSION}-base-ubuntu24.04
 
 ENV DEBIAN_FRONTEND noninteractive
 
+# Install necessary packages
 RUN apt-get update && apt-get install -y --no-install-recommends \
     wget \
     curl \
     software-properties-common \
+    python3 \
+    python3-pip \
+    nodejs \
+    npm \
     && apt-get clean \
     && rm -rf /var/lib/apt/lists/*
 
 WORKDIR /workspace
 
-RUN wget https://github.com/cdr/code-server/releases/download/v4.92.2/code-server_4.92.2_amd64.deb \
-    && dpkg -i code-server_4.92.2_amd64.deb \
-    && rm code-server_4.92.2_amd64.deb
+# Install code-server
+RUN curl -fsSL https://code-server.dev/install.sh | sh
+
+# Install some common VS Code extensions
+RUN code-server --install-extension ms-python.python \
+    && code-server --install-extension ms-toolsai.jupyter
 
+# Install Jupyter
+RUN pip3 install jupyter
+
+# Expose port for code-server
 EXPOSE 8080
-CMD ["code-server", "--bind-addr", "0.0.0.0:8080"]
+
+# Start code-server
+CMD ["code-server", "--bind-addr", "0.0.0.0:8080", "--auth", "none"]