corrected includes in KMeans