Added another test, minor modifications. Added __init__.py file.