Add reboot time test and keep screen on when websocket connected