Minor changes

Minor changes to UI.