diff --git a/models/User.py b/models/User.py
index e4f2384..13f9f72 100644
--- a/models/User.py
+++ b/models/User.py
@@ -1,8 +1,8 @@
 from datetime import datetime
 
 from flask_login import UserMixin
-from db.SQLiteClient import create_user, get_user, get_user_by_email, get_habits, delete_user, update_user, \
-    get_habitLists
+from db.SQLiteClient import create_user, get_user, get_user_by_email, delete_user, update_user, \
+    get_habitLists, get_heatmap_value
 
 
 class User(UserMixin):
@@ -43,3 +43,10 @@ class User(UserMixin):
             habitLists.append(habitList)
 
         return habitLists
+
+    def get_heatmap(self):
+        heatmap = []
+        for day in range (0, 27):
+            value = get_heatmap_value(self.id, day)
+            heatmap.append(value)
+        return heatmap
\ No newline at end of file