quick tidy up