Update agent.py

docstring updated