cleaning up the ui