Update comments to be docstrings