add callback to setTheme as well