Add wave class