Fast Fourier Transform implementation in python with latex equations