Add docstrings to TorAgent