Oi, Diego. Tudo bem?
Seu código ficou excelente, está correto sim! Parabéns pela resolução do desafio :)
Aproveito para deixar uma dica de como você pode adicionar seu código aqui no fórum. Uma maneira é clicando no botão </> na barra superior da mensagem, após clicar, dois conjuntos de três crases aparecerão e você pode colar seu código dentro! Dessa forma:
Qualquer dúvida, estamos sempre por aqui.
Bons estudos!
Isso mesmo! Parabéns!