Updated the wavelet notebook a bit more