Update docstrings

A previous change added an argument to several methods without
updating their docstrings to describe it.

Change-Id: I10c753a2d935c0a4ae87ff9258c91f1530d398db