added __init__.py to work as a module