Bug fix. Kmeans initialization