Update trax.py