change to new category segmentation