Update agent.go