change the function to select the leading leptons and change the categories