DOC: Fix plots in docs