Update display.hs