Add UI and code to BuilderGeneratorExtension