diff --git a/setup.sh b/setup.sh
index 888e638..843ef27 100644
--- a/setup.sh
+++ b/setup.sh
@@ -7,6 +7,7 @@ sudo usermod -a -G kvm $USER
 
 # Switch to testing
 sudo cp google-chrome.list /etc/apt/sources.list.d/
+sudo cp vscode.list /etc/apt/sources.list.d/
 sudo cp -f sources.list /etc/apt/
 wget -q -O - https://dl.google.com/linux/linux_signing_key.pub | sudo apt-key add -
 wget -qO- https://packages.microsoft.com/keys/microsoft.asc | gpg --dearmor > packages.microsoft.gpg
@@ -26,11 +27,11 @@ sudo apt install -y mesa-vulkan-drivers libglx-mesa0:i386 mesa-vulkan-drivers:i3
 
 
 # Tools
-sudo apt install -y zsh nala lsd fonts-font-awesome neofetch mc tmux batcat curl plocate apt-file libvirt-clients sassc libglib2.0-dev-bin curl wget
-sudo apt update
+sudo apt install -y zsh nala lsd fonts-font-awesome neofetch mc tmux curl plocate apt-file libvirt-clients sassc libglib2.0-dev-bin curl wget
 sudo apt-file update
 sudo usermod -a -G libvirt $USER
 
+
 # Flatpak
 sudo sudo apt install flatpak 
 sudo apt install gnome-software-plugin-flatpak
@@ -87,6 +88,7 @@ sh -c "$(curl -fsSL https://raw.githubusercontent.com/ohmyzsh/ohmyzsh/master/too
 git clone https://github.com/zsh-users/zsh-autosuggestions ${ZSH_CUSTOM:-~/.oh-my-zsh/custom}/plugins/zsh-autosuggestions
 git clone https://github.com/zsh-users/zsh-syntax-highlighting.git ${ZSH_CUSTOM:-~/.oh-my-zsh/custom}/plugins/zsh-syntax-highlighting
 cp .zshrc $HOME
+chsh -s /bin/zsh
 
 # Steam
 sudo apt install -y steam-installer
@@ -96,7 +98,8 @@ sudo apt install -y steam-installer
 dconf load / < dconf.toml
 
 # Applications
-sudo apt install google-chrome-stable code
+sudo apt install google-chrome-stable 
+sudo apt install code
 
 # Clean up
 sudo apt autoremove