pylustrator generates unused code?

[mmagnuski]

Hi, first of all - thanks for a great package! I only used it for a while now but I already see how it will make my life easier. :)

The issue I'm reporting is just a small nuisance: during my testing pylustrator seemed to add unused code at the begining of its generated output, namely:

plt.figure(1).ax_dict = {ax.get_label(): ax for ax in plt.figure(1).axes}
import matplotlib as mpl

The mpl import is unused in the generated code below. The ax_dict is also not used anywhere below. If it is used internally by pylustrator it might be useful to add a comment to this line.