Ricerca elemento in Array Multidimensionale

[Aires]

Ciao a tutti, avrei un problemino con la ricerca negli Array Multidimensionali.
Inanzitutto mi spiego, ho un Array di nome Elementi questo array conterrà un ID, un FLAG ed un VALUE, nell'array registrerò più ID con i rispettivi valori.

Utilizzando questa sintassi stampo correttamente a video gli elementi

$elemento_ricerca = "campo_x";
    
foreach($Elementi AS $key => $val){    
  foreach ($val AS $key2 => $val2){
    if($val2 == $elemento_ricerca){ $msg = 'Trovato --> ['.$key.']['.$key2.']'; }
  }
} 

Con questo sistema recupero le indicazioni per 'richiamare' il campo ricercato.
Il problema sorge ora, volevo tramutare questa parte di codice in una funzione in maniera tale da richiamare ogni volta un 'elemento di ricerca' diverso e farmi inviare tramite return le indicazioni, tuttavia così facendo non funge, idee?

function test($elemento_ricerca){
  foreach($Elementi AS $key => $val){    
    foreach ($val AS $key2 => $val2){
    if($val2 == $elemento_ricerca){ $msg = ' Trovato --> ['.$key.']['.$key2.']'; }
     }
   }            
  return $msg;
}
        
echo test("campo_a");
echo test("campo_b");
echo test("campo_c");