improve segmentation