Add braces to if statements that were missing