tidy up the main UI