Part-2: Segmentation with feedback