Adding reboot functionality