Added command-line argument to build.sh