Added stay_on_screen function