Updated the comments for more readability