adding python code for fft visualization