I think the roots are little lost in this version, but other than that it's a really nice improvement. Maybe use a colour adjusted version of the tree distribution mask softened up with the Smoothing Filter shader, could be used to mask out some grass around tree bases or "covered" area where light doesn't reach too much.