diff --git a/src/ubuntu/install/edge/install_edge.sh b/src/ubuntu/install/edge/install_edge.sh
index ad3a104..62d935b 100644
--- a/src/ubuntu/install/edge/install_edge.sh
+++ b/src/ubuntu/install/edge/install_edge.sh
@@ -5,7 +5,7 @@ CHROME_ARGS="--password-store=basic --no-sandbox --disable-gpu --user-data-dir -
 
 apt-get update
 
-EDGE_BUILD=$(curl -q https://packages.microsoft.com/repos/edge/pool/main/m/microsoft-edge-dev/ | grep href | grep .deb | sed 's/.*href="//g'  | cut -d '"' -f1 | tail -1)
+EDGE_BUILD=$(curl -q https://packages.microsoft.com/repos/edge/pool/main/m/microsoft-edge-dev/ | grep href | grep .deb | sed 's/.*href="//g'  | cut -d '"' -f1 | sort --version-sort | tail -1)
 
 wget -q -O edge.deb https://packages.microsoft.com/repos/edge/pool/main/m/microsoft-edge-dev/$EDGE_BUILD
 apt-get install -y ./edge.deb
diff --git a/src/ubuntu/install/vs_code/install_vs_code.sh b/src/ubuntu/install/vs_code/install_vs_code.sh
index b8a00c1..49dc63d 100644
--- a/src/ubuntu/install/vs_code/install_vs_code.sh
+++ b/src/ubuntu/install/vs_code/install_vs_code.sh
@@ -1,7 +1,8 @@
 #!/usr/bin/env bash
 set -ex
 wget -q https://update.code.visualstudio.com/latest/linux-deb-x64/stable -O vs_code_amd64.deb
-dpkg -i vs_code_amd64.deb
+apt-get update
+apt-get install -y ./vs_code_amd64.deb
 sed -i 's#/usr/share/code/code#/usr/share/code/code --no-sandbox##' /usr/share/applications/code.desktop
 cp /usr/share/applications/code.desktop $HOME/Desktop
 chmod +x $HOME/Desktop/code.desktop
@@ -14,3 +15,10 @@ apt-get install -y python3-setuptools \
                    python3-venv \
                    python3-virtualenv \
                    maximus
+
+# Cleanup
+apt-get autoclean
+rm -rf \
+    /var/lib/apt/lists/* \
+    /var/tmp/* \
+    /tmp/*