DOC: add a couple plt.close('all') to avoid warning
DOC: fix warning + some stylistic corrections