Added support for other branches than master