Add support for branches