changed last line in kmeans.py