Update demo_segmentation.m