Update display_lib.py

smarter