Finished k-means clustering in python