Add updates to build script.