Code cleanup: method comments to docstring