Minor changes to display buttons