Add segmentation.py