Add brackets to if clause