Just a bit of code re-styling