Update agent.rb