Memory trace in prefrontal cortex: theory for the cognitive switch