Added reboot feature