Fala Francisco, tudo certo!
Muito legal a sua solução! Obrigado por compartilhar com a gente aqui no fórum! Utilizar enumerações para resolver esse tipo de problema é bastante comum e faz bastante sentido!
Parabéns pela dedicação e lembre-se que você pode sempre contar com o fórum para tirar suas dúvidas e compartilhar sua evolução, adoramos acompanhar sua trajetória!
Um grande abraço e bons estudos!