Add __init__.py required for the import in setup
Fix entry point in setup.py