python 2d fft example