diff --git a/install.sh b/install.sh
index da8dbcb..869e7a2 100644
--- a/install.sh
+++ b/install.sh
@@ -7,7 +7,10 @@ lsblk -f
 echo "----------"
 echo "Which drive do we want to use for this installation?"
 echo "For example /dev/sda or /dev/nvme0n1"
-read driveName
+
+echo ""
+read -p "Enter your drive choice: " driveName
+echo ""
 
 ## Download Disko file
 cd /tmp
@@ -19,11 +22,17 @@ sudo sed -i "s#/dev/sda#$driveName#g" /tmp/disko-config.nix
 # Step 2: Partitioning the drive used for the installation
 
 ## Run Disko to partition the disk
-sudo nix --experimental-features "nix-command flakes" run github:nix-community/disko/latest -- --mode destroy,format,mount /tmp/disko-config.nix
+sudo nix --experimental-features "nix-command flakes" run github:nix-community/disko/latest -- --mode destroy,format,mount /tmp/disko-config.nix --yes-wipe-all-disks
 
 ## Generate Nix configuration
 sudo nixos-generate-config --no-filesystems --root /mnt
 
+## Copies over the disko file for running `nixos-install`
+sudo mv /tmp/disko-config.nix /mnt/etc/nixos
+
+## Confirm that partitions are mounted
+lsblk -f
+
 ## Downloads and places the predefinded generic flake to use
 curl https://gitlab.com/ahoneybun/nix-configs/-/raw/main/flake.nix > flake.nix; sudo mv -f flake.nix /mnt/etc/nixos/
 
@@ -43,15 +52,9 @@ Which device are you installing to?
 EOF
 
 echo ""
-read -p "Enter your choice: " hostChoice
+read -p "Enter your device choice: " hostChoice
 echo ""
 
-## Copies over the disko file for running `nixos-install`
-sudo mv /tmp/disko-config.nix /mnt/etc/nixos
-
-## Workaround for `no free space` error
-sudo nix-collect-garbage
-
 if [ $hostChoice = 1 ]; then
    curl https://gitlab.com/ahoneybun/nix-configs/-/raw/main/hosts/x86_64/vm/configuration.nix > vm.nix; sudo mv -f vm.nix /mnt/etc/nixos/
    sudo nixos-install --flake /mnt/etc/nixos#vm
@@ -68,6 +71,7 @@ elif [ $hostChoice = 3 ]; then
 
 elif [ $hostChoice = 0 ]; then
    curl https://gitlab.com/ahoneybun/nix-configs/-/raw/main/flake.nix > flake.nix; sudo mv -f flake.nix /mnt/etc/nixos/
+   curl https://gitlab.com/ahoneybun/nix-configs/-/raw/main/desktops/gnome.nix > gnome.nix; sudo mv -f gnome.nix /mnt/etc/nixos/
    sudo nixos-install --flake /mnt/etc/nixos#nixos
 
 fi