diff --git a/images/Dockerfile.pytorch b/images/Dockerfile.pytorch
index 5ffe3ec..bc2f3b3 100644
--- a/images/Dockerfile.pytorch
+++ b/images/Dockerfile.pytorch
@@ -2,3 +2,11 @@ FROM quay.io/jupyter/pytorch-notebook:cuda12-2024-04-22
 COPY jupyter-ai.yml environment.yml
 RUN conda env update --file environment.yml
 
+USER root
+RUN curl -L https://ollama.com/download/ollama-linux-amd64 -o /usr/local/bin/ollama && chmod +x /usr/local/bin/ollama
+RUN curl -fsSL https://code-server.dev/install.sh | sh && rm -rf .cache
+RUN git config --system pull.rebase false && \
+    git config --system credential.helper 'cache --timeout=30000' && \
+    echo '"\e[5~": history-search-backward' >> /etc/inputrc && \
+    echo '"\e[6~": history-search-forward' >> /etc/inputrc
+
diff --git a/images/jupyter-ai.yml b/images/jupyter-ai.yml
index 8c98215..1a6f808 100644
--- a/images/jupyter-ai.yml
+++ b/images/jupyter-ai.yml
@@ -4,4 +4,7 @@ channels:
 dependencies:
   - jupyterlab-myst
   - jupyter-ai
+  - jupyter-tensorboard-proxy 
+  - jupyter-resource-usage
+  - jupyter-vscode-proxy