Putting useful imports into __init__.py