Update power_spectra.py