Enrichment rescues contextual discrimination deficit associated with immediate shock