adding kmeans wip