a little tidy up