Allow setting of custom branch

Closes #15