ENH small edits to plotting functions