Update screen_setup.py