added better way to do muon reco