Update plot_eigenworms.py