Improving __init__.py files