Got the custom UI elements to work.