Update align.py

Fixes #104, for consistancy's sake
Merge pull request #106 from andrewolaughlin/master

Update align.py