Slight tidy up