Update plot_kazr.py