Fix kmeans_2d and gmm example