Just add the theme support in register(), add test for it.