Remove module imports in __init__