diff --git a/README.md b/README.md
index 2fc56b8..1f0e07f 100644
--- a/README.md
+++ b/README.md
@@ -55,7 +55,7 @@ https://nixos.org/manual/nixos/stable/index.html#sec-installation-booting-networ
 ## Start the installer
 
 ```sh
-sh <(curl -L https://gitlab.com/ahoneybun/nyxi-installer/-/raw/main-pbp/install.sh)
+sh <(curl -L https://gitlab.com/ahoneybun/nyxi-installer/-/raw/main/install.sh)
 ```
 
 The following will happen:
@@ -70,7 +70,7 @@ The following will happen:
 - Creates a root partition in the LVM.
 - Installs systemd-boot
 
-### ARM64 (Pinebook Pro) - main-pbp branch 
+### ARM64 (Pinebook Pro) - main-pbp branch
 
 - Clear partition table for `/dev/***`. 
 - Creates a GPT partition table for `/dev/***`.
diff --git a/Screenshots/.gitkeep b/Screenshots/.gitkeep
new file mode 100644
index 0000000..e69de29
diff --git a/Screenshots/Screenshot_from_2023-01-08_00-21-43.png b/Screenshots/Screenshot_from_2023-01-08_00-21-43.png
new file mode 100644
index 0000000..55a0570
Binary files /dev/null and b/Screenshots/Screenshot_from_2023-01-08_00-21-43.png differ
diff --git a/Screenshots/nixos-pantheon.png b/Screenshots/nixos-pantheon.png
new file mode 100644
index 0000000..dfb7ae7
Binary files /dev/null and b/Screenshots/nixos-pantheon.png differ
diff --git a/Screenshots/pinebook-pro_nixos.png b/Screenshots/pinebook-pro_nixos.png
new file mode 100644
index 0000000..b029355
Binary files /dev/null and b/Screenshots/pinebook-pro_nixos.png differ
diff --git a/install.sh b/install.sh
index 8de8c49..136f3d0 100644
--- a/install.sh
+++ b/install.sh
@@ -21,10 +21,6 @@ echo +1G     # Set +1G as last sector.
 echo n       # Create new partition (for root).
 echo         # Set default partition number.
 echo         # Set default first sector.
-echo -4G     # Set last sector.
-echo n       # Create Swap partition
-echo         # Set default partition number.
-echo         # Set default first sector.
 echo         # Set last sector.
 echo t       # Change partition type.
 echo 1       # Pick first partition.
@@ -91,17 +87,17 @@ read desktopChoice
 # Update the second command to the file name that matches your DE/WM .nix file
 
 if [ $desktopChoice = 1 ]; then
-   curl https://gitlab.com/ahoneybun/nix-configs/-/raw/main/plasma.nix > plasma.nix; sudo mv -f plasma.nix /mnt/etc/nixos/
+   curl https://gitlab.com/ahoneybun/nix-configs/-/raw/main/desktops/plasma.nix > plasma.nix; sudo mv -f plasma.nix /mnt/etc/nixos/
    sudo sed -i "10 i \           ./plasma.nix" /mnt/etc/nixos/configuration.nix
 else
 
 if [ $desktopChoice = 2 ]; then
-   curl https://gitlab.com/ahoneybun/nix-configs/-/raw/main/gnome.nix > gnome.nix; sudo mv -f gnome.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 sed -i "10 i \           ./gnome.nix" /mnt/etc/nixos/configuration.nix
 fi
 
 if [ $desktopChoice = 3 ]; then
-   curl https://gitlab.com/ahoneybun/nix-configs/-/raw/main/pantheon.nix > pantheon.nix; sudo mv -f pantheon.nix /mnt/etc/nixos/
+   curl https://gitlab.com/ahoneybun/nix-configs/-/raw/main/desktops/pantheon.nix > pantheon.nix; sudo mv -f pantheon.nix /mnt/etc/nixos/
    sudo sed -i "10 i \           ./pantheon.nix" /mnt/etc/nixos/configuration.nix
 fi