Add Wave class