Using kmeans clustering from scipy.cluster