small update to ui