Don’t import plot in __init__