Add branch support