Modified rules to make tree look better