Switched to mini-batch kmeans