update unit tests, update comment.