build.sh and Dockerfile updated