minor fixes to UI