The goal of this protocol is to perform a behavioral assay such as the attentional set shifting task (AST) to assess prefrontal cortex-mediated cognitive flexibility in mice.