Updated 2d segmentation chapter