Add comment checking