rework screen on/off code