Update transition.py