diff --git a/dice_stats.py b/dice_stats.py
index 83a99cb..7ecdef2 100644
--- a/dice_stats.py
+++ b/dice_stats.py
@@ -18,13 +18,30 @@ class FiveDice:
             if face != 1:
                 return False
         return True
+    
+    def is_three_of_a_kind(self):
+        if sum([d.face == self.dice[0].face for d in self.dice]) == 3:
+            return True
+        if sum([d.face == self.dice[1].face for d in self.dice]) == 3:
+            return True
+        if sum([d.face == self.dice[2].face for d in self.dice]) == 3:
+            return True
+        return False
+    
+    def is_four_of_a_kind(self):
+        if sum([d.face == self.dice[0].face for d in self.dice]) == 4:
+            return True
+        if sum([d.face == self.dice[1].face for d in self.dice]) == 4:
+            return True
+        return False
+        
 
 dice = FiveDice()
 successes = 0
 trials = 1000000
 for trial in tqdm(range(trials)):
     dice.roll()
-    if dice.all_ones():
+    if dice.is_four_of_a_kind():
         successes += 1
 
 print(successes/trials)