Update wavelets.py