Super naive but fast kmeans