Removed if __main__