minor fixes to screen.py