Modified docstring, cosmetic changes