Update __init__.py

    plt.legend(plist)
↓
    plt.legend([plist[n]])