This was an intriguing problem. Given more time it would be great to add more here.
As it is, this problem is really easy, but it was a very interesting part of the lab.
Maybe in the long run we could make a separate lab and do more with higher dimensional FFT's.