refactored to make a single lepton tree/selection