Put back docstring as comment