Add braces for if/then blocks