moved imports to the top in __init__.py