Update to include auto reboot