diff --git a/root/etc/cont-init.d/95-apt-get b/root/etc/cont-init.d/95-apt-get
new file mode 100644
index 0000000..5b1a21c
--- /dev/null
+++ b/root/etc/cont-init.d/95-apt-get
@@ -0,0 +1,3 @@
+#!/usr/bin/with-contenv bash
+
+apt-get update
diff --git a/root/etc/cont-init.d/98-zsh b/root/etc/cont-init.d/98-zsh
index 284050e..24a4774 100644
--- a/root/etc/cont-init.d/98-zsh
+++ b/root/etc/cont-init.d/98-zsh
@@ -2,20 +2,20 @@
 
 # Install ZSH
 echo "**** Installing zsh ****"
-apt-get update && apt-get install -y \
-  zsh
+apt-get install -y \
+    zsh
 
 # Install oh-my-zsh if not installed already
 if [ -d /config/.oh-my-zsh ]; then
-  echo "**** Oh My Zsh is already installed. You can update it via \"omz update\" in terminal. ****"
+    echo "**** Oh My Zsh is already installed. You can update it via \"omz update\" in terminal. ****"
 elif [ -d /.oh-my-zsh ]; then
-  echo "**** Installing Oh My Zsh. You can update it via \"omz update\" in terminal. ****"
-  mv /.oh-my-zsh /config/.oh-my-zsh
-  cp /config/.oh-my-zsh/templates/zshrc.zsh-template /config/.zshrc
+    echo "**** Installing Oh My Zsh. You can update it via \"omz update\" in terminal. ****"
+    mv /.oh-my-zsh /config/.oh-my-zsh
+    cp /config/.oh-my-zsh/templates/zshrc.zsh-template /config/.zshrc
 else
-  echo "**** It looks like Oh My Zsh was manually deleted after install. Please recreate the code-server container to reinstall Oh My Zsh. ****"
+    echo "**** It looks like Oh My Zsh was manually deleted after install. Please recreate the code-server container to reinstall Oh My Zsh. ****"
 fi
 
 chown -R abc:abc \
-  /config/.oh-my-zsh \
-  /config/.zshrc
+    /config/.oh-my-zsh \
+    /config/.zshrc