Adding in the custom element modal