code clean and add comment.