Add back the theming function