Add support for arguments to before/after/around