Docstring fix
Updated changes