more small fixes to ui