Add class for FSM. Not compilable for some reason