Added some comments to kmeans