Updated the ui