Got plot_example.py working again