Update display.py

Fixed #38