Alterado o enum e melhoria do codigo