Tidy up a bit ;)