Add __init__.py to tests folder and move main logic into main() (#12)

Add __init__.py to tests folder and move main logic into main() (#12) #24