Code clean and update comment.