refine TreebankReader

fix tree doctest