Got rid of redundant `main` method, put it all in __main__