Add plot_clusters function.