Add surprisal computation.