Written STFT function, need to test it