Docstring and a few small changes.