Update wavelet.py