prettify arguments list
Add note on saving figures and docstring example