Clean up docstring (add references)