updated code to fix comment.