Modified multi_kmeans.py