added update_screen function