[Sugestão] Simplificação do código

Eu fiz meu código assim, alcancei os mesmo resultados da professora. Nao sei se esta correto, mas tentei deixar o código com menos linhas

//toca som
function tocaSom(seletorAudio) {
   const elemento = document.querySelector(seletorAudio);

   elemento === null ? console.log('Elemento nao encontrado'): elemento.play();
}
const listaDeTeclas = document.querySelectorAll('.tecla');

for (let contador = 0; contador < listaDeTeclas.length; contador++) {
    const tecla = listaDeTeclas[contador];
    const instrumento = tecla.classList[1];
    const idAudio = `#som_${instrumento}`;
    tecla.onclick = () => tocaSom(idAudio);

    tecla.onkeydown = (evento) => {

        evento.code == 'Space' || evento.code == 'Enter' ? tecla.classList.add('ativa') : '';
        tecla.onkeyup = () => tecla.classList.remove('ativa');
    }
}