Adding a unit test