diff --git a/test/homework2/main.py b/test/homework2/main.py
new file mode 100644
index 0000000..450cd98
--- /dev/null
+++ b/test/homework2/main.py
@@ -0,0 +1,14 @@
+import numpy as np
+
+a = np.arange(2 * 3 * 4).reshape(2, 3, 4)
+
+b = np.arange(12).reshape(2, 6)
+
+c = a + b.reshape(2, 3, 4)
+
+c_flattened = c.flatten(copy=True)
+
+print("a\n", a)
+print("b\n", b)
+print("c\n", c)
+print("c_flattened\n", c_flattened)