Added a new method for returning additional information after the segmentation