add __name__ == __main__