Added curly braces

Oops