Move docstring to the right function