Corrigido um pequeno bug e limpeza de código