Add balance field in the Wallet object