change getBalance in wallet