Same as before but with plot_wham