Added Ben's tree visualization