Add docstring to plot method