small changes to Wallet