Se agrega la lógica de tratamientos