minor fixes to ui