Whitespace tidy up