changing comment to docstring