Allow function arguments to with and each

Via @mcdan
Fixes #239