Fala Matteus, tudo bem?
Boa!! Ficou excelente a solução
Uma dica, quando for postar algum código aqui no fórum, utilize crases antes e depois do código para que ele fique formatado para que todos consigam ver, assim:
Abraços e bons estudos :D