Add support for selecting a custom branch