Improved image segmentation