Update tsm.py