Updated the comment on the class