diff --git a/install.sh b/install.sh
index 5e86d12..d3a0295 100755
--- a/install.sh
+++ b/install.sh
@@ -598,4 +598,12 @@ runScript() {
     done
 }
 
+umountAndClean() {
+    swapoff /mnt/swapfile
+    rm /mnt/swapfile
+    umount -R /mnt
+    swapoff /dev/sda2
+    parted -s /dev/sda mklabel gpt
+}
+
 runScript "$@"