Update pytms.py