Update align.py

corrected a typo, added a final "else" case to alignment function