Fixed code for style according to Wenderlich's style guide.