Add class to proper theme