Improve traveling salesman demo