Update traitAlign.py