Olá Matheus, tudo bem com você?
Peço desculpas pela demora em obter retorno.
Muito obrigado por compartilhar sua solução conosco aqui no fórum, ficou muito boa! Isolar esses trechos de código dentro de funções facilita muito a legibilidade e o entendimento do que o programa faz, né? Gostei bastante de como seu programa ficou enxuto e coeso.
Lembre-se que você pode sempre contar com o fórum para tirar suas dúvidas e compartilhar a sua evolução, adoramos saber o que você está estudando!
Um grande abraço e bons estudos.