Add braces to 'if' statement