diff --git a/week06/average-squares-example/average_squares/squares.py b/week06/average-squares-example/average_squares/squares.py
index e7d64a1..5f1c636 100644
--- a/week06/average-squares-example/average_squares/squares.py
+++ b/week06/average-squares-example/average_squares/squares.py
@@ -1,5 +1,6 @@
 """Computation of weighted average of squares."""
 
+from argparse import ArgumentParser
 
 def average_of_squares(list_of_numbers, list_of_weights=None):
     """ Return the weighted average of a list of values.
@@ -51,11 +52,15 @@ def convert_numbers(list_of_strings):
 
 
 if __name__ == "__main__":
-    numbers_strings = ["1","2","4"]
-    weight_strings = ["1","1","1"]        
+    parser = ArgumentParser(description="Weighted square average of a list of values")
+
+    parser.add_argument('numbers', nargs='+', default=["1","2","4"])
+    parser.add_argument('--weights', '-w', nargs='+', default=["1","1","1"])
+
+    arguments = parser.parse_args()  
     
-    numbers = convert_numbers(numbers_strings)
-    weights = convert_numbers(weight_strings)
+    numbers = convert_numbers(arguments.numbers)
+    weights = convert_numbers(arguments.weights)
     
     result = average_of_squares(numbers, weights)