feature addition in kmeans.py