Added __init__.py

Better structuring of __init__.py to ease imports.