add comment and class the function