Probability learning to avoid shock