Move some comments to docstring