add - finished unit tests