fixes to ui