#1 Add a unit test