Improves docs to be more like numpydoc