Style tidy up