Small update to build script