Updated color segmentation