Add if/else