Fixed example __main__