Update build with new functionality