Customize the build to build the docker image