diff --git a/btrfs-test.sh b/btrfs-test.sh
index 7218404..2a1905a 100644
--- a/btrfs-test.sh
+++ b/btrfs-test.sh
@@ -54,14 +54,6 @@ IFS=' '
 # Create new variable of selection.
 array=($dev)
 dev="${array[0]}"
-szmn="${array[1]%?}"
-
-# Setting partition size.
-#echo "Your system partition has a size of $szmn GB."
-#echo "What size should your root volume have (GB) ?"
-#echo "The rest will be used for your home volume."
-#read szrt
-#echo
 
 echo "2. ENCRYPTION"
 echo
@@ -156,7 +148,6 @@ sgdisk -Z /dev/$dev &&
 echo
 
 # Creating boot partition.
-#szmn="${array[1]%?}"
 echo "Creating boot partition of 256 MB."
 sgdisk -n 1:0:+256M /dev/$dev &&
 echo
@@ -165,7 +156,7 @@ echo
 sgdisk -t 1:ef00 /dev/$dev &&
 
 # Creating system partition.
-echo "Creating system partition of $szmn GB."
+echo "Creating system partition."
 sgdisk -n 2:0:0 /dev/$dev
 echo
 
@@ -327,7 +318,7 @@ linux   /vmlinuz-linux-zen
 initrd  /initramfs-linux-zen.img
 options cryptdevice=/dev/$main:main
 options root=/dev/mapper/main rw
-options rootflags=subvol=root
+options rootflags=subvol=@root
 options lang=de locale=de_DE.UTF-8
 options init=/usr/lib/systemd/systemd
 #options loglevel=3
@@ -339,7 +330,8 @@ title   Arch Linux Fallback
 linux   /vmlinuz-linux
 initrd  /initramfs-linux-fallback.img
 options cryptdevice=/dev/$main:main
-#options root=/dev/mapper/main-root rw
+options root=/dev/mapper/main rw
+options rootflags=subvol=@root
 options lang=de locale=de_DE.UTF-8
 options init=/usr/lib/systemd/systemd
 EOF
@@ -373,9 +365,12 @@ echo
 # Unmounting and rebooting.
 echo "Unmounting system."
 umount /mnt/boot &&
-umount /mnt/ &&
+umount /mnt &&
 echo "Done."
 echo
 
-echo "Rebooting"
-reboot
+clear
+
+echo "Installation finished!"
+echo "Press [Enter] to shut down, then remove the installation media and start your system."
+poweroff