Adding fft plot to ext.py