Add branch flexibility