Add ability to change wave type