Add if __main__