Updated the code for comments.