include extra tree's + peter's code