Modified StateManager to use the new Response format