Added support for different branches