criado classes que faltavam e ajustado a logica