Olá, Diego, como vai?
O código que você desenvolveu cumpre perfeitamente o objetivo do exercício, utilizando de forma correta a entrada de dados via console e a aplicação da lógica matemática através de uma função. É interessante notar que você optou por isolar o cálculo em um método dedicado, o que é uma excelente prática de organização e reutilização de código em projetos reais.
Parabéns pelo empenho em resolver a atividade e por compartilhar sua solução com a comunidade do fórum, continue com essa dedicação nos próximos desafios.
Conte com o apoio da comunidade Alura na sua jornada. Abraços e bons estudos!