Update drawwaveform.py