Fix docstring placement