Some minor fixes to the UI.