codedex-io
diff --git a/‎3-control-flow/16_sorting_hat_2.py
Copy file name to clipboardExpand all lines: 3-control-flow/16_sorting_hat_2.py
+8-4Lines changed: 8 additions & 4 deletions b/‎3-control-flow/16_sorting_hat_2.py
Copy file name to clipboardExpand all lines: 3-control-flow/16_sorting_hat_2.py
+8-4Lines changed: 8 additions & 4 deletions
@@ -77,11 +77,15 @@
 print("Hufflepuff: ", hufflepuff)
 print("Slytherin: ", slytherin)
 
-if gryffindor >= ravenclaw and gryffindor >= hufflepuff and gryffindor >= slytherin:
+# Bonus Part
+
+most_points = max(gryffindor, ravenclaw, hufflepuff, slytherin)   # We'll learn about the max() function in Chapter 6
+
+if gryffindor == most_points:
   print('🦁 Gryffindor!')
-elif ravenclaw >= hufflepuff and ravenclaw >= slytherin:
+elif ravenclaw == most_points:
   print('🦅 Ravenclaw!')
-elif hufflepuff >= slytherin:
+elif hufflepuff == most_points:
   print('🦡 Hufflepuff!')
 else:
-  print('🐍 Slytherin!')
+  print('🐍 Slytherin!')