Adding a unit test or two.