Wrap function arguments too. Fixes #106.