Fix missing parameter from docstring

There were missing parameters in docstring after latest changes. This patch is to include those docstring.