Electrical stimulation improved cognitive deficits associated with traumatic brain injury in rats