Add segmentation test