Create screen.py