Add braces

It's good practice, y'know?