add support for specifying branch during upgrade