Added flag to keep screen on.