add comment logic