removed redundant code from plot_fft_2d.py