Modify __main__.py to allow running from wheel