diff --git a/baselines/common/tf_util.py b/baselines/common/tf_util.py
index 2699035..452871c 100644
--- a/baselines/common/tf_util.py
+++ b/baselines/common/tf_util.py
@@ -165,6 +165,10 @@ def function(inputs, outputs, updates=None, givens=None):
     outputs: [tf.Variable] or tf.Variable
         list of outputs or a single output to be returned from function. Returned
         value will also have the same shape.
+    updates: [tf.Operation] or tf.Operation
+        list of update functions or single update function that will be run whenever
+        the function is called. The return is ignored.
+        
     """
     if isinstance(outputs, list):
         return _Function(inputs, outputs, updates, givens=givens)