Update kmeans.cpp