Add the theme wrapper