add more features for segmentation