Update tsn.py