adding kmeans.py