Minor changes to docstring based on review suggestions