Continue assim!!!
Boa tarde!
Sua resolução ficou muito boa!
Só sugiro nas próximas vezes utilizar as tags de código </>, um ícone presente no topo da caixa de texto, para o seu código ficará mais legivel.
Parabéns pelo empenho nos estudos e continue praticando.
Para que mais pessoas consigam visualizar a sua solução proposta, é recomendado que você marque este tópico como solucionado. Assim pode ajudar ainda mais pessoas :)
Bons estudos!