fixed comments, added models back ingto traffic.py