Update waves.py