Update screen.py

Added some comments to show bits to be removed/modified