Update display_lib.py

error catch on shape size