I think the block `if __name__ == '__main__':` is not necessary.