diff --git a/amundsen_common/models/table.py b/amundsen_common/models/table.py
index 5c6b36f..592c728 100644
--- a/amundsen_common/models/table.py
+++ b/amundsen_common/models/table.py
@@ -33,6 +33,19 @@ class Meta:
         register_as_scheme = True
 
 
+@attr.s(auto_attribs=True, kw_only=True)
+class Badge:
+    badge_name: str = attr.ib()
+    category: str = attr.ib()
+    badge_type: str = attr.ib()
+
+
+class BadgeSchema(AttrsSchema):
+    class Meta:
+        target = Badge
+        register_as_scheme = True
+
+
 @attr.s(auto_attribs=True, kw_only=True)
 class Watermark:
     watermark_type: Optional[str] = None
diff --git a/setup.py b/setup.py
index 0b95675..acb1d39 100644
--- a/setup.py
+++ b/setup.py
@@ -5,7 +5,7 @@
 
 setup(
     name='amundsen-common',
-    version='0.3.6',
+    version='0.4.0',
     description='Common code library for Amundsen',
     long_description=open('README.md').read(),
     long_description_content_type='text/markdown',