Travelling Salesman Problem in Python