Able to pass in InitialBranchState to TraversalAStar