Import class in __init__