Cambiando la clase porque no me gustaba asi
Añadiendo test