UI additions
improvements to ui