Fixed display.py, had forgotten if __name__ line.
Fixed display.py again, forgot an =.