improvements to ui