ENH: Updating to get new segmentation options from ANTs