Update to build script