Fixed styling of code