Improved shouldBranch accuracy