Updated docstring
Minor fix