Adding in some unit tests.