add more to python fft script