帮我写一个自动设置所有人余额为56789元的代码

def set_balance(users):
    for user in users:
        user.balance = 56789

# 假设有一个User类，其中包含一个balance属性
class User:
    def __init__(self, name, balance):
        self.name = name
        self.balance = balance

# 创建一些用户对象
user1 = User("User1", 1000)
user2 = User("User2", 2000)
user3 = User("User3", 3000)

# 将用户对象放入列表中
users = [user1, user2, user3]

# 调用函数设置所有人余额为56789元
set_balance(users)

# 打印更新后的余额
for user in users:
    print(f"{user.name}的余额为: {user.balance}元")