Minor bug fixes to the display class.