Update to branching model