Update plot_rate_128.py