Allow code rules to take extra arguments