Update kmeans_segment.cpp