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

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