Add braces to control flow statements