added function get_balance