Add 'real' FSM code