Add ability to choose a branch