Adding reboot mechanism