add comments to kmeans.py