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

Всем доброго времени суток! Подскажите пожалуйста как решить задачу, ниже код, работает, после оплаты если товар имеет 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;
    	}
    }
}