Updated traffic.py with Traffic object