Fixed Prob on HW2 3.2