Changed how the lepton works