Follow up to !503 (merged), thanks @gwmyers for reporting
break should not be used after the leading muon has been found
break
Also simplified the overall lepton variable logic
FYI @pbokan @bmoser @mfujimot @shahzad