Adapted the sample for modified branch handling in Shippable