State Machine In Python