Simplify and add umount/reboot at the end