added diffusion to plot_bz_phase_plot.py