fixup! Refactor collocation logic to collogation class, create probability calculations