Added the return to the reboot function