added arguments to get_balance