added a unit test. More to come