Update plot_cluster.py