Update kmeans.py