Add comment for the newly added class