more updates to ui