improve consistency of w_plot with other plots in chapter