Add branch choice to upgrade job.