I am using in-vivo two-photon calcium imaging to study the neural circuitry underlying cognitive flexibility during a set-shifting task.