small tweaks to ui