def update():
    for t in range(10):
        s = env.reset()
        while True:
            env.render()
            a = 1
            (s, r, done) = env.step(a)
            if done:
                break
def update():
    for t in range(10):
        s = env.reset()
        while True:
            env.render()
            a = 1
            (s, r, done) = env.step(a)
            if done:
                break