Create wave_fun.py

Basic demo of using waves.py