diff --git a/unnecessarymath.py b/unnecessarymath.py
index b1c11fedb9f57a2e8d01ed499152ef7ae2dc68a0..0985cd871a346cb459146b9089777029fa28bd11 100644
--- a/unnecessarymath.py
+++ b/unnecessarymath.py
@@ -8,12 +8,18 @@ def sqrt(x):
 
     >>> sqrt(4)
     2.0
-
     >>> sqrt(1.)
     1.0
     """
     return np.sqrt(x)
 
 def square(x):
-    """Evaluate the square of a number"""
+    """
+    Evaluate the square of a number
+
+    >>> square(2)
+    4
+    >>> square(4.)
+    16.
+    """
     return x * x