Added a square function to the foo file

Author: markbasham

src/demo/foo.py

@@ -22,3 +22,8 @@ def mul(a, b):
 
 def div(a, b):
     return a/b
+
+
+def square(a):
+    return a*a
+