Olá, William! Tudo bem?
Excelente solução, parabéns!
Aproveito para deixar uma sugestão de como você pode adicionar seu código aqui no fórum, a melhor maneira é clicando no botão </> na barra superior da mensagem, após clicar, dois conjuntos de três crases aparecerão e você pode escrever seu código dentro. Dessa forma:
Continue se empenhando. Qualquer dúvida, estamos à disposição!
Abraços e bons estudos.