Add drawing of waves