Add sys.stdout.flush