Corrigindo bugs e limpando o código