Tidy up the ui