add option for setting the branch.

[drmowinckels]

Since "master" branch is being renames "main" for new repositories, it seems like a good idea to be able to set the branch name in the function call.

[benmarwick]

Could we perhaps go a bit further and set branch = "main" also, what do you think?