diff --git a/entry.sh b/entry.sh
index eb410db..0a90615 100755
--- a/entry.sh
+++ b/entry.sh
@@ -67,7 +67,7 @@ function first_stage(){
 	mount ${DEVICE}1 $TARGET/boot
 	echo mount ${DEVICE}1 $TARGET/boot
 	genfstab -U $TARGET | tee $TARGET/etc/fstab
-	UUID=$(lsblk -lpo NAME,UUID | grep ${DEVICE}2 | awk '{print $1}')
+	UUID=$(lsblk -lpo NAME,UUID | grep ${DEVICE}2 | awk '{print $2}')
 	echo cryptlvm UUID=${UUID} none luks > $TARGET/etc/crypttab
 }
 
@@ -77,5 +77,9 @@ function second_stage(){
 	passwd
 	echo -n > /etc/motd
 	apt install -y linux-image-amd64 grub2 cryptsetup btrfs-progs lvm2 locales tzdata keyboard-configuration console-common
+	mkdir /snap
+	btrfs subvolume snapshot / /snap/$(date +%Y-%m-%d_basesystem)
+	apt install -y task-mate-desktop mate-desktop-environment-extra
+	
 }