I'd prefer this to be split up into the if/else statement