Added support for branches