Update agent_module.py