clean up the ui