added line to keep the screen on