Minor updates to the UI