Register the custom UI for stats

Temporarily add this, will come back and fix