Update agent.py

updates to docstrings