A: add unit test