diff --git a/Dockerfile b/Dockerfile
index b7b40f0..7fa2f75 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -2,7 +2,9 @@
 FROM pangeo/pangeo-notebook:2025.01.24
 
 COPY environment.yml /tmp/environment.yml
+COPY postBuild /tmp/postBuild
 
 RUN mamba env update --prefix /srv/conda/envs/notebook --file /tmp/environment.yml
+RUN bash /tmp/postBuild
 
 RUN ls
diff --git a/postBuild b/postBuild
index 7938055..0605b7d 100644
--- a/postBuild
+++ b/postBuild
@@ -4,9 +4,14 @@ set -euo pipefail
 
 # Define packages
 declare -a packages=( ms-python.python svelte.svelte-vscode openai.chatgpt anthropic.claude-code )
+target_user="${NB_USER:-jovyan}"
 
 # Install them
 for package in "${packages[@]}"
 do
-  echo code-server --install-extension "${package}"
+  if [ "$(id -u)" -eq 0 ] && [ "${target_user}" != "root" ]; then
+    su - "${target_user}" -c "code-server --install-extension ${package} --force"
+  else
+    code-server --install-extension "${package}" --force
+  fi
 done
\ No newline at end of file