Simplified if/else