2d clustering with kmeans