added support for custom branch
added support for custom branch