refactor: comments for WenWei's version