Update cli.py

Author: TheComputationalCore

probability_calculator/cli.py

@@ -5,34 +5,50 @@
 
 def main():
     parser = argparse.ArgumentParser(
-        description="Probability Calculator — Monte-Carlo simulation for drawing balls from a hat."
+        description=(
+            "Probability Calculator — Monte-Carlo simulation for drawing "
+            "colored balls from a hat."
+        )
     )
 
     parser.add_argument(
         "--hat",
         nargs="+",
-        help="Colors and counts, e.g. red=3 blue=2",
+        help="Colors and counts: e.g. red=3 blue=2 green=6",
     )
+
     parser.add_argument(
         "--expect",
         nargs="+",
-        help="Expected ball counts, e.g. red=2 blue=1",
+        help="Required minimum colors: e.g. red=2 green=1",
+    )
+
+    parser.add_argument(
+        "--draw",
+        type=int,
+        help="Number of balls to draw.",
+    )
+
+    parser.add_argument(
+        "--experiments",
+        type=int,
+        help="Number of Monte-Carlo experiments.",
     )
-    parser.add_argument("--draw", type=int, help="Number of balls drawn")
-    parser.add_argument("--experiments", type=int, help="Number of experiments")
 
     args = parser.parse_args()
 
+    # Must pass all required arguments
     if not (args.hat and args.expect and args.draw and args.experiments):
         parser.print_help()
         return
 
-    # Convert hat arguments
+    # Parse hat contents
     hat_kwargs = {}
     for item in args.hat:
         color, count = item.split("=")
         hat_kwargs[color] = int(count)
 
+    # Parse expected outcome
     expected = {}
     for item in args.expect:
         color, count = item.split("=")