modify traffic utils.py