Update transit.py