Restructured, started playing with numpy fft