Add FlowStateMap support to get REPL tutorial running again