add comprehension test