Added reboot button and mode selection