Added an option to keep screen on when connected.