Add braces for correct formatting