Small edits to build script.