Add comment to classes