Added wave functionality