Update build script for customization