Changed to update input segmentation