Adding in Segmentation tests