Change scala KMeans now that map as last op in interation works