Update the build.sh for build