Update ts_tools.py