Get wallet working

This PR adds some tests and fixes a small handful of bugs. The tests are not very thorough, but enough to catch the major errors that we were having.

Over the next day or two I plan on expanding out the wallet tests a bit more.