diff --git a/fundamentals/python/square.py b/fundamentals/python/square.py
new file mode 100644
index 0000000..6de28a1
--- /dev/null
+++ b/fundamentals/python/square.py
@@ -0,0 +1,4 @@
+# Now you have to write a function that takes an argument and returns the square of it.
+
+def square(n):
+    return n * n # return the square of n
\ No newline at end of file