Add FSM class