Allow a function body to be passed to stub_command