Add more tree code.