Add logging to __init__.py file