Add curly braces around if statement