Поддержка WooCommerce Автоматическая смена ролей после оплаты

  • Всем доброго времени суток! Подскажите пожалуйста как решить задачу, ниже код, работает, после оплаты если товар имеет id 151, то удаляется старая роль: customer и записывается новая роль: um_custom_role_6. Вопрос, как правильно добавить еще одно условие, что если товар имеет id 203, то удаляется старая роль: customer и записывается новая роль: um_custom_role_4 ?

    
    add_action( 'woocommerce_order_status_completed', 'change_role_on_purchase' );
    function change_role_on_purchase( $order_id ) {
        $order = wc_get_order( $order_id );
        $items = $order->get_items();
    
        $products_to_check = array( '151' );
    
        foreach ( $items as $item ) {
            if ( $order->user_id > 0 && in_array( $item['product_id'], $products_to_check ) ) {
            	$user = new WP_User( $order->user_id );
    
            	// Change role
            	$user->remove_role( 'customer' );
            	$user->add_role( 'um_custom_role_6' );
    
                // Exit the loop
                break;
        	}
        }
    }