added __init__.py and more documentation