Помоги исправить код

Всем привет.

Необходимо на определенные категории добавить в магазине кнопку Демоверсии после корзины.

Есть такой код у меня.

// Add demo preview button to single product page

 

function isa_after_add_to_cart_button() {

 

    global $post;

 

    $demoslug = $post->post_name;

 

     $demourl = get_bloginfo('url').'/preview?theme='.$demoslug.'';

 

        $demotitle = esc_attr($post->post_title);

 

    echo '<a href="'.$demourl.'" title="'.$demotitle.'" class="sc-button demo-preview" rel="nofollow" target="_blank">Демо-сайт</a>';

 

}

add_action('woocommerce_after_add_to_cart_button','isa_after_add_to_cart_button');

Но данный код добавляет кнопку на все товары.

Ранее я делал дурацкий код для того что бы убрать (скрыть) кнопку с ненужных категорий и делал это так:

// Hide shop items for services category

function services_remove_woocommerce(){

	global $product, $post;

	$target_category = 'services';



	// Remove shop items from the Single Product page

	if ( is_product() ) {

		$terms = wp_get_post_terms( $post->ID, 'product_cat' );

		foreach ( $terms as $term ) $categories[] = $term->slug;

		if ( in_array( $target_category, $categories ) ){

    ?>

    <style type="text/css">

	a.sc-button.demo-preview {

    display: none;

    }

    </style>

    <?php

    }

	}

}

add_action( 'wp_head', 'services_remove_woocommerce' );

Подскажите пожалуйста как просто установить кнопку на некотоыре категории. Кнопку нужно по сути добавить на товар из осного раздела и на его подкатегории.