work on the fourier transform ndarray code