add reboot feature