Add missing reboot function