Make the arguments array optional
Add a 'return_arg' option to specify an argument to return
Add documentation for return_arg