Add muon association check