Add comment and docstring