Add if/else helpers