Minor updates to display