Update docstring in plot_events.py