Update build functionality