Update align.py