Olá, Mauricio, como vai?
O seu código está correto, cumpre exatamente o que o enunciado pede e a lógica usada está clara e objetiva. Você estruturou bem as condições e usou a função de forma adequada para tornar o programa reutilizável.
Você fez um bom trabalho e está no caminho certo. Continue compartilhando seus exercícios, isso ajuda bastante no aprendizado e pode inspirar outros colegas também. O fórum está à disposição sempre que precisar.
Conte com o apoio da comunidade Alura na sua jornada. Abraços e bons estudos!