Improve the power spectrum plots