tidy up the main UI +2