Make comment into docstring as suggested by @mhvk