adding EM steps to KMeans