Minor changes to FFT notebook