Update build with new UI