Added __init__.py for tests to work