Update tsp.py