Python: Added exception to display.py