minor update to ui