Add comment functionality