slight fixes to ui