Update a comment and a docstring