Oi, Evandro! Tudo bem?
Pode sim! Seu código ficou bem legível e claro, parabéns pela resolução!
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 sempre por aqui.
Bons estudos.