Update k-means-clustering.py