change to loop over jets instead