Custom Modeling Element Tray with search capability

The out of the box element tray provided by BPMN.io does not provide capabilities like search or element sorting/classification. It would be useful to be able to search through the elements to find the correct element, especially for element sub-types