Remove extra comment, fix tiny bug
Add missing __init__.py for build