Tidy up
Another tidy up