Added an option to keep the screen on