Enhance cognitive flexibility