Update fft.py