Add __init__.py for easier importing. Fix test case.