fixing plotting issues with watershed segmentation