Add code to update category for product