add an option to plot U-matrix in plot_graph