Would be really nice and convenient if there was an option to allow code to wrap in the function node