Wrote some unit tests.