Updating code with comments.