Add braces to if statement