Move the branch setup to the script