updated to work with WooCommerce 2.0