add more tests to wallet