Update wrap to correctly return wrapped function arity
Merge pull request #520 from arcseldon/wrap

Update wrap to correctly return wrapped function arity