Improved AgentMemory