diff --git a/get_num_square.py b/get_num_square.py
new file mode 100644
index 0000000..b4d4cda
--- /dev/null
+++ b/get_num_square.py
@@ -0,0 +1,11 @@
+import os
+
+num = os.environ.get("INPUT_NUM")
+if num:
+    try:
+        num = int(num)
+    except Exception:
+        exit("ERROR: the INPUT_NUM provided ({num}) is not an integer")
+    print(num**2)
+else:
+    num = 1