Add docstrings to the ArgumentGenerator class