Move docstring to module level

Docstring still needs to be updated.