Update matlab_code/surprise.m