Minor changes to Display