minor fix
semi-finel changes to ui