Removed importing classes in __init__