Fix docstring into numpy style