ADD unit test for previous work