more updates to the ui