Pass the branch name to code checkout script

* Pass the branch name to code checkout script