Added support for specifying a branch to track