Minor updates

Minor updates to the UI.