Travelling Salesman Problem Code Python