Hi bellenot,
I am getting these branches from the tree then make their Lorentz vector (in the step you mentioned).