Add basic testing for __main__.py