Added leptons to small_tree