minor change in kmeans.py