Moved docstrings to the correct place