Update traffic.py