Update tpt.py