small improvements to ui