Add option to pass branch name as flag.