Slight update to the build script