updated traffic.py to use variables instead of hard coded values