plt.scatter(X[y_kmeans == 1, 0], X[y_kmeans == 1, 1],