Wrote a bunch of unit tests