TSP in Python