Extend estraverse rules to support JSX