Added woocommerce support - functions.php