Small cleanup

You don't need to name your file `__main__.py` for the `if __name__ == '__main__'` thing to work.