Added UI elements for assembly and root type selection