Made some edits to the display UI.