Add comment about using arg as the return and link to docs