Fold logging stuff into __init__.py