diff --git a/max submission/6_9_HW.py b/max submission/6_9_HW.py
index 97570d8..22725d4 100644
--- a/max submission/6_9_HW.py	
+++ b/max submission/6_9_HW.py	
@@ -94,15 +94,24 @@ if __name__ == "__main__":
 
 # question 3:
 
+def calculate_square(x):
+    square = x ** 2
+    return square
+
+def calculate_cube(x):
+    cube = x ** 3
+    return cube
+
 def main():
-    x = 5
-    y = 2
-    breakpoint()
-    result = x + y  # Line of interest
+    number = 5
+    breakpoint()  # Set breakpoint here
+    square_result = calculate_square(number)
+    cube_result = calculate_cube(number)
+    print("Square:", square_result)
+    print("Cube:", cube_result)
 
-    print(result)
-
-main()
+if __name__ == "__main__":
+    main()