Update fsm.py