State management for Flows