Add Missing imports in __init__