@commands.command()
@commands.cooldown(1, 86400, commands.BucketType.user)
async def work(self, ctx):
    await self.open_account(ctx.author)
    user_id = str(ctx.author.id)
    user_jobs = self.get_user_jobs(user_id)
    
    # Calculate earnings for each job
    for job_name in user_jobs:
        job_info = self.jobs[job_name]
        earnings = job_info['base_pay']
        
        # Check for bonus
        if random.random() < job_info['bonus_chance']:
            bonus_earned = job_info['bonus_amount']
            earnings += bonus_earned

@commands.command()
async def gamble(self, ctx, amount=None):
    await self.open_account(ctx.author)
    users = await self.get_bank_data()
    
    if amount == "all":
        amount = users[str(user.id)]["wallet"]
    elif isinstance(amount, str) and "%" in amount:
        percentage = int(amount.replace("%", ""))
        amount = int(users[str(user.id)]["wallet"] * (percentage / 100))
    
    # Coin flip - heads or tails
    result = "heads" if random.randint(1, 2) == 1 else "tails"
    win = result == "heads"

async def add_balance(self, user_id, amount):
    """Add balance to a user's account (admin command)"""
    users = await self.get_bank_data()
    
    # Convert user_id to string for consistency
    user_id = str(user_id)
    
    # Create account if user doesn't exist
    if user_id not in users:
        users[user_id] = {}
        users[user_id]["wallet"] = 0
        users[user_id]["bank"] = 0
    
    # Add amount to wallet
    users[user_id]["wallet"] += amount