Add keep screen on feature.