tidy up ui