Update docstring

Reformat docstring