Update account_balance.py