Fixed up the styling of the code.