Add support for Total theme