Moved to milk for faster/better kmeans