Figured out how to use branch callback