Add brackets to if statements