Minor style clean up