Update segmentation.js