Bom dia, Regina! Tudo bem?
Gosto bastante da estrutura do Switch Case para resolver esses tipos de problema, e você conseguiu aplicar esse da forma correta e deixou o projeto com uma cara mais fácil de entender o fluxo.
Uma curiosidade é que, nas versões mais atuais do Java, a partir da versão 17, o Switch Case recebeu uma nova versão um pouco mais polida, então recomendo que dê uma conferida:
No mais, bons estudos!