Fixing typeo in fft.py in power_spectrum method, was looking at wrong elements