Update white_noise_plot.py