Move main method to __main__.py