add a simple implementation of kmeans