minor fixes to the UI