diff --git a/src/main/java/com/kamco/cd/training/postgres/entity/ModelDatasetEntity.java b/src/main/java/com/kamco/cd/training/postgres/entity/ModelDatasetEntity.java
index d78c038..a5e98f7 100644
--- a/src/main/java/com/kamco/cd/training/postgres/entity/ModelDatasetEntity.java
+++ b/src/main/java/com/kamco/cd/training/postgres/entity/ModelDatasetEntity.java
@@ -38,6 +38,12 @@ public class ModelDatasetEntity {
   @Column(name = "container_cnt")
   private Long containerCnt;
 
+  @Column(name = "waste_cnt")
+  private Long wasteCnt;
+
+  @Column(name = "land_cover_cnt")
+  private Long landCoverCnt;
+
   @ColumnDefault("now()")
   @Column(name = "created_dttm")
   private OffsetDateTime createdDttm;