DOC: Improve docstrings in plot module