Improved the update-reboot script