few tweaks to ui