add comments on unit test