Update check_balance.cpp