More story
Fix prefix_probability problem