screen.py fully working create function