Update align.py

Fixes #104, for consistancy's sake