filled up update balance TODOs with function call to update balance