Adding an image I forgot to add and removed print statements I had added in display.