Added a couple unit tests.