diff --git a/medium/count_total_number_of_colored_cells.py b/medium/count_total_number_of_colored_cells.py
new file mode 100644
index 0000000..16f115e
--- /dev/null
+++ b/medium/count_total_number_of_colored_cells.py
@@ -0,0 +1,8 @@
+# https://leetcode.com/problems/count-total-number-of-colored-cells
+
+class Solution:
+    def coloredCells(self, n: int) -> int:
+        if n == 1:
+            return 1
+        return (n - 1) ** 2 + n ** 2
+