Add comments to segmentation script.