add new feature to retrieved balance.
TODO:refactory stuff