Corrigindo o codigo e os testes