[WIP] Added kmeans

We should probably support kmeans...