slight tidy up