The attentional set shifting task: a measure of cognitive flexibility in mice.