add check_balance