Move import in __init__ file