Revise comment
Add proper indentation and closing braces to if statement