Update alignment.py

Fixed bug - extract_wave_new