Changed single line comments to docstrings and re-tested