Added full training data, and switched to FFT representation